代码随想录DAY21|二叉树part08

669.修剪二叉搜索树

题目链接

669. 修剪二叉搜索树 - 力扣（LeetCode）

题目描述

给你二叉搜索树的根节点 root ，同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树，使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即，如果没有被移除，原有的父代子代关系都应当保留)。 可以证明，存在 唯一的答案 。

所以结果应当返回修剪好的二叉搜索树的新的根节点。注意，根节点可能会根据给定的边界发生改变。

代码思路

遍历查找不符合要求的节点并删除，而且由于二叉搜索树的特性，如果一个节点小于low，则其左子树都不符合条件，只对右子树进行处理即可，如果一个节点大于high，则其右子树都不符合条件，只对左子树进行处理即可。

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(root==NULL) return NULL;
        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;

    }
};

本来有疑惑的地方在于：好像只对root的左子树和右子树进行处理，并没有对root进行处理，但是看了前面的三个if发现，当root不满足条件时，会直接返回其返回条件的子树，故已经经过筛选与处理了。

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

题目链接

108. 将有序数组转换为二叉搜索树 - 力扣（LeetCode）

题目描述

给你一个整数数组 nums ，其中元素已经按 升序 排列，请你将其转换为一棵平衡二叉搜索树。

平衡二叉树 是指该树所有节点的左右子树的深度相差不超过 1。

代码思路

找到中间的节点命为根节点root，依次向左向右，找到中间节点为该节点的左子树和右子树，直至只剩两个或一个节点。

别说，自己写真是绕进去了，不知道怎么迭代计算中间值，还想着通过while进行进行，不知道该方法是不是可以，但是我在写的时候遇到了阻碍，有点无以为继的感觉，遂看了标准答案。

最重要的时创建了个迭代函数，就像之前的二分查找一般，用[left, right]区间进行迭代：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    TreeNode* travelsal(vector<int>& nums,int left,int right){
        if(left>right) return NULL;
        int mid=(left+right)/2;
        TreeNode* root=new TreeNode(nums[mid]);
        root->left=travelsal(nums,left,mid-1);
        root->right=travelsal(nums,mid+1,right);
        return root; 
    }
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return travelsal(nums,0,nums.size()-1);
    }
};

只能说，记住二分查找的代码写作方式记住，区间的迭代方法！

538.把二叉搜索树转换为累加树

题目链接

538. 把二叉搜索树转换为累加树 - 力扣（LeetCode）

题目描述

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

• 节点的左子树仅包含键 小于 节点键的节点。
• 节点的右子树仅包含键 大于 节点键的节点。
• 左右子树也必须是二叉搜索树。

代码思路

顺序就是右中左的遍历顺序依次相加，就是实现遍历相加，但是遍历好像又不太会写了，看了看代码随想录的思路，看到里面的迭代函数需要前一个节点的值，然后就在对中间节点处理时直接加上上一节点的值，然后发现循环的终止条件是NULL，所以返回的第一个节点是NULL，所以会报错，故而应当是直接用int存储值，改正后的代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    int pre=0;
    void rmlorder(TreeNode* cur){
        if(cur==NULL) return;
        rmlorder(cur->right);
        cur->val+=pre;
        pre=cur->val;
        rmlorder(cur->left);
        return;
    }
public:
    
    TreeNode* convertBST(TreeNode* root) {

        rmlorder(root);
        return root;
    }
};