程式碼隨想錄演算法訓練營,9月17日 | 669. 修剪二叉搜尋樹,108.將有序陣列轉換為二叉搜尋樹,538.把二叉搜尋樹轉換為累加樹

漪欢酒發表於2024-09-17

669. 修剪二叉搜尋樹
題目連結:669. 修剪二叉搜尋樹
文件講解︰程式碼隨想錄(programmercarl.com)
影片講解︰修剪二叉搜尋樹
日期:2024-09-17

想法:節點為空返回空,值在中間時,繼續遞迴左右兩邊,小於時遞迴右子樹,大於時遞迴左子樹
Java程式碼如下:

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if(root == null) return null;
        if(root.val < low){
            return trimBST(root.right, low, high);
        }
        if(root.val > high){
            return trimBST(root.left, low, high);
        }
        root.left = trimBST(root.left, low, high);
        root.right = trimBST(root.right, low, high);
        return root;
    }
}

108.將有序陣列轉換為二叉搜尋樹
題目連結:108.將有序陣列轉換為二叉搜尋樹
文件講解︰程式碼隨想錄(programmercarl.com)
影片講解︰將有序陣列轉換為二叉搜尋樹
日期:2024-09-17

想法:跟構造二叉樹同樣的思路,用下標表示遞迴的範圍,很簡單。
Java程式碼如下:

class Solution {
    public TreeNode sortedArrayToBST1(int[] nums, int left, int right){
        if(left > right) return null;
        int mid = left + (right - left) / 2;
        TreeNode node = new TreeNode(nums[mid]);
        node.left = sortedArrayToBST1(nums, left, mid - 1);
        node.right = sortedArrayToBST1(nums, mid + 1, right);
        return node;
    }
    public TreeNode sortedArrayToBST(int[] nums) {
        TreeNode root = sortedArrayToBST1(nums, 0, nums.length - 1);
        return root;
    }
}

538.把二叉搜尋樹轉換為累加樹
題目連結:538.把二叉搜尋樹轉換為累加樹
文件講解︰程式碼隨想錄(programmercarl.com)
影片講解︰把二叉搜尋樹轉換為累加樹
日期:2024-09-17

想法:反中序遍歷累加
Java程式碼如下:

class Solution {
    int sum;
    public TreeNode convertBST(TreeNode root) {
        sum = 0;
        convertBST1(root);
        return root;
    }

    public void convertBST1(TreeNode root) {
        if (root == null) {
            return;
        }
        convertBST1(root.right);
        sum += root.val;
        root.val = sum;
        convertBST1(root.left);
    }
}

相關文章