2021-02-13 二叉树累加和 修剪二叉搜索树

题目1 把二叉搜索树转换为累加树

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

示例 1：

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

思路

对于二叉搜索树，针对每个节点，大于或等于 node.val 的值之和，即该结点右边所有结点的和。因此可以反向中序遍历树，按照右根左的顺序进行遍历，并将结点和累加

解1 递归

反向递归遍历右子树，并通过一个全局变量，记录遍历的右边节点的和，然后覆盖。

class Solution {
    int sum = 0;
public:
    TreeNode* convertBST(TreeNode* root) {
        if(!root) return NULL;
        if(root->right) convertBST(root->right);
        sum += root->val;
        root->val = sum;
        if(root->left) convertBST(root->left);
        return root;
    }
};

解2 morris反向遍历

class Solution {
public:
    TreeNode* convertBST(TreeNode* root) {
        int sum = 0;
        TreeNode* cur = root;
        TreeNode* pre = NULL;
        while(cur != NULL)
        {
        	//右结点为空
            if(NULL == cur->right)
            {
            	//加和
                sum += cur->val;
                //覆盖
                cur->val = sum;
                cur = cur->left;
            }
            else
            {
            	//找右子树的最左结点
                pre = cur->right;
                while(pre->left != NULL && pre->left != cur)
                    pre = pre->left;
                //若为空，左孩子等于cur
                if(pre->left == NULL)
                {
                    pre->left = cur;
                    cur = cur->right;
                }
                //若左孩子为cur
                else
                {
                	//加和覆盖
                    sum += cur->val;
                    cur->val = sum;
                    //恢复原状
                    pre->left = NULL;
                    cur = cur->left;
                }
            }
        }
        return root;
    }
};

题目2 修建二叉搜索树

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

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

示例 1：

输入：root = [1,0,2], low = 1, high = 2
输出：[1,null,2]

示例 2：

输入：root = [3,0,4,null,2,null,null,1], low = 1, high = 3
输出：[3,2,null,1]

解 递归

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
    	//若结点值大于最大值，则该节点及其右子树均不符合条件，只处理左子树，并返回左子树处理结果的根
        if(root->val > high)
        {
            if(root->left) return trimBST(root->left, low , high);
        }
        //若结点值小于最小值，则该节点及其左子树均不符合条件，只处理右子树，并返回右子树处理结果的根
        else if(root->val < low)
        {
            if(root->right) return trimBST(root->right, low, high);
        }
        //若结点值符合条件，递归处理左子树、右子树，返回该节点 
        else
        {
            if(root->left) root->left = trimBST(root->left, low, high);
            if(root->right) root->right = trimBST(root->right, low, high);
            return root;
        }
        return NULL;
    }
};