算法学习 | day20/60 修剪二叉树/将有序数组转成二叉搜索树/把二叉树转成累加树/二叉树总结

 一、题目打卡

        1.1 修剪二叉树（答案思路更好）

        题目链接：. - 力扣（LeetCode）

class Solution {
private:
    TreeNode* trimLow(TreeNode* root, int i){
        if(!root) return nullptr;
        root->right = trimLow(root->right,i);
        root->left = trimLow(root->left,i);
        if(!root->left && !root->right && root->val < i ){
            delete root;
            return nullptr;
        }
        if(root->right && root->val < i){
            // TreeNode* tmp = root->right;
            // delete root;
            // return tmp;
            TreeNode* tmp = root;
            root = root->right;
            // delete tmp;
            return root;
        }
        return root;
    }

    TreeNode* trimHigh(TreeNode* root, int i){
        if(!root) return nullptr;
        root->right = trimHigh(root->right,i);
        root->left = trimHigh(root->left,i);
        if(!root->left && !root->right && root->val > i ){
            // delete root;
            return nullptr;
        }
        if(root->left && root->val > i){
            // TreeNode* tmp = root->left;
            // delete root;
            // return tmp;
            TreeNode* tmp = root;
            root = root->left;
            // delete tmp;
            return root;
        }
        return root;
    }
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        root = trimLow(root, low);
        root = trimHigh(root, high);
        return root;
    }
};

        自己写的有点冗余，写的过程中遇到的坑：首先是这个不能删除节点，删除的话代码会报错，应该是删除了头结点导致的；其次是如果按照我这么写，我尝试了中序处理，后续处理，最后是一个测试案例出错让我反应过来，我这样处理的话，其实本质上是不停剪叶子节点，所以应该是后续遍历，但是我其实很好奇我最开始思路为什么做不出来，也就是用中序遍历，但是更换第一个递归的顺序，这样本质上是缩小区间的方式，但是没写出来（down.jpg)，看了答案发现这个思路有点麻烦，所以感觉也暂时没有尝试的必要了，看二刷的时候有没有精力再做一次吧；最后就是如果是我那样写的话（不要这么写，我不想再简化了），记住一定要在调用两次递归函数前面加上返回值，让root = trimBST，不然会出错。

        相对答案采用的是前序处理的方法，感觉其实更直观，答案处理的那种情况其实我也都考虑到了，只是确实没有写出来这种思路，然后我看了解析，自己写了一次：

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

        1.2 将有序数组转换为二叉搜索树 

        题目链接：. - 力扣（LeetCode）

class Solution {
private:
    TreeNode* recur(vector<int>& nums, int i, int j){
        // if(j < i || i == j) return new TreeNode(nums[i]);
        if(i>j) return nullptr;
        if(i==j) return new TreeNode(nums[i]);
        int m = (i + j) / 2;
        TreeNode* root = new TreeNode(nums[m]);
        root->left = recur(nums,i,m-1);
        root->right = recur(nums,m+1,j);
        return root;
    }
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        int i = 0, j = nums.size() - 1;
        return recur(nums,i,j);
    }
};

        1.3 把二叉搜索树转换为累加树

        题目链接：. - 力扣（LeetCode）

class Solution {
public:
    int pre = 0;
    TreeNode* convertBST(TreeNode* root) {
        if(!root) return nullptr;
        convertBST(root->right);
        // cout << root->val << " "; 
        // if(pre == 0){
        //     pre = root->val;
        //     return root;
        // }
        root->val = pre + root->val;
        pre = root->val;
        convertBST(root->left);
        return root;
    }
};

        利用二叉搜索树的特性，直接从右边开始深度优先搜索就可以了，这里注意的是，对于首个元素，因为声明的前一个指针，并把值设置为0，其实没有必要再特殊化处理了。

二、总结

        内容来自：代码随想录        

        详细的内容已经总结的很好了，最后借用一下图和最终的总结：

        在二叉树题目选择什么遍历顺序是不少同学头疼的事情，我们做了这么多二叉树的题目了，Carl给大家大体分分类。

• 涉及到二叉树的构造，无论普通二叉树还是二叉搜索树一定前序，都是先构造中节点。

• 求普通二叉树的属性，一般是后序，一般要通过递归函数的返回值做计算。

• 求二叉搜索树的属性，一定是中序了，要不白瞎了有序性了。

注意在普通二叉树的属性中，我用的是一般为后序，例如单纯求深度就用前序，二叉树：找所有路径(opens new window)也用了前序，这是为了方便让父节点指向子节点。

所以求普通二叉树的属性还是要具体问题具体分析。

二叉树专题汇聚为一张图：