第二十三天:修建二叉树的思想还要重点掌握!

本文介绍了三种针对二叉搜索树的操作:1)TrimaBinarySearchTree,通过递归实现剪枝操作;2)ConvertSortedArraytoBinarySearchTree,利用分治策略将有序数组转换为平衡二叉搜索树;3)ConvertBSTtoGreaterTree,转换二叉搜索树,使其每个节点值为原值加上所有小于它的节点值之和。

1.(重点!!!)Trim a Binary Search Tree

这题代码不难,但确实挺难想到的。尤其是用递归去跳过剪枝的操作,太强了!

class Solution {
private:
    TreeNode* dfs(TreeNode* root, int low, int high){
        if(!root) return root;

        if(root->val < low){
            TreeNode* right = trimBST(root->right, low, high); 
            return right;
        }else if(root->val > high){
            TreeNode* left = trimBST(root->left, low, high); 
            return left;
        }
        root->left = dfs(root->left, low, high);
        root->right = dfs(root->right, low, high);
        return root;
    }
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        return dfs(root, low, high);
    }
};

2.Convert Sorted Array to Binary Search Tree

我本来的想法是slice整个vector,但后来发现可以做,但是太麻烦。不如借助divide and conquer思想,把这个先搞了。

class Solution {
private:
    TreeNode* dfs(vector<int>& nums, int l, int r){
        if(l == r){
            return new TreeNode(nums[l]);
        }else if(l > r){
            return nullptr;
        }
        int mid = (l + r) / 2;
        TreeNode* root = new TreeNode(nums[mid]);
        root->left = dfs(nums, l, mid-1);
        root->right = dfs(nums, mid+1, r);
        return root;
    }
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        int len = nums.size()-1;
        return dfs(nums, 0, len);
    }
};

3.Convert BST to Greater Tree

纸老虎题,本质上只是在考察遍历顺序罢了。

class Solution {
private:
    int sum = 0;
    void dfs(TreeNode* root){
        if(!root) return;
        dfs(root->right);
        root->val += sum;
        sum = root->val;
        dfs(root->left);
    }
public:
    TreeNode* convertBST(TreeNode* root) {
        dfs(root);
        return root;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值