代码随想录第十八天|669. 修剪二叉搜索树108.将有序数组转换为二叉搜索树538.把二叉搜索树转换为累加树

最新推荐文章于 2024-10-09 22:28:58 发布

王禄DUT

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量130

点赞数 1

分类专栏：代码随想录文章标签：算法 leetcode 深度优先

本文链接：https://blog.csdn.net/m0_73596392/article/details/142316258

版权

代码随想录专栏收录该内容

32 篇文章 0 订阅

订阅专栏

修剪二叉搜索树：

返回值代表修剪之后的二叉树的根节点

所以对于NULLreturn NULL

对于不符合的返回右子树和左子树的修剪结果

对于符合的修剪左右子树最后返回修剪完的子树

通过接住返回值来达到删除结点的目的

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);
          else if(root->val>high)
              return trimBST(root->left,low,right);
          else{
            root->left=trimBST(root->left,low,high);
            root->right=trimBST(root->right,low,high);
            return root;
          }
    }
};

class Solution {
public:
    void dfs(TreeNode *&root,vector<int> nums){
        if(nums.empty())
            return;
        else{
            int mid=nums.size()/2;
            root=new TreeNode(nums[mid]);
            dfs(root->left,vector<int>(nums.begin(),nums.begin()+mid));
            dfs(root->right,vector<int>(nums.begin()+mid+1,nums.end()));
        }
    }
    TreeNode* sortedArrayToBST(vector<int>& nums) {
           TreeNode *root=NULL;
           dfs(root,nums);
           return root;
    }
};

class Solution {
public:
    int sum=0;
    //左根右是顺序 右根左是倒叙
    void dfs(TreeNode *&root){
        if(root==NULL){
            return;
        }else{
            dfs(root->right);
            sum+=root->val;
            root->val=sum;
            dfs(root->left);
        }

    }
    TreeNode* convertBST(TreeNode* root) {
          dfs(root);
          return root;
    }
};