LeetCode1038：从二叉搜索树到更大和树

一、题目描述

给出二叉搜索树的根节点，该二叉树的节点值各不相同，修改二叉树，使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。下图为实例：

输入：[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]

二、解题思路

由于题目规定是二叉搜索树，那么由二叉搜索树的性质，可知根节点小于任意右子树，大于任意左子树。这里最右端的叶节点一定是最大值，最左端叶节点一定是最小值。
思路：

• 从最右端的节点开始依次往上回溯
• 将当前节点加上直接右子树的值；
• 若当前节点有左子树，则左子树加上当前节点。

下图是根节点与右子树调整步骤：

三、代码实现

通过递归实现，这里详细解释递归过程：下面有两个函数

• bstToGst主函数
• help：具体负责递归，int型为了在跳出递归条件时保存现场信息(用return xx保存)。

int help(TreeNode* &root,int val){
    if (root==nullptr) {
        return 0;
    }
    int temp = 0;
    if (root->right!=nullptr) {
        temp=help(root->right, val);
        root->val+=temp;//使右树的父节点加上右子树的值
    }
    else{//最右的叶节点最大，只能等于本身，所以第一个递归传入的val必须为0
        root->val+=val;
        
    }
    temp=root->val;
    if (root->left!=nullptr) {
        //当前节点有左子树时(比如这里是原来的6，已变为21)
        temp=help(root->left, root->val);//将当前节点的左子树加上当前节点(5+21），所以必须将val变成当前节点的值(21)
    }
    return temp;//保存现场信息(当前已修正节点的值)
}

TreeNode* bstToGst(TreeNode* root){
    help(root, 0);
    return root;
}