剑指 Offer II 054. 所有大于等于节点的值之和

中序遍历

1. 中序遍历右子树

2. 中序遍历根节点

3. 中序遍历左子树

/**
 * 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* convertBST(TreeNode* root) {
        // 反向后序遍历，优先遍历右子树，然后根节点，然后左子树  右-左+当前值
        if(root==NULL){
            return root;
        }
        int sum = 0;
        // 第一步对二叉树进行求和
        sumTree(root,sum);
        // 修改节点的值
        //modifyTree(root,sum);
        return root;
    }
    void  sumTree(TreeNode* root,int& sum){
        // 如果是根节点，那么应该是本身加上父节点与父节点右子树的总和
        if(root->left==nullptr&&root->right==nullptr){
            sum+=root->val;
            root->val = sum;
            return ;
        }
        // 获取右子树的总和
        if(root->right!=nullptr){
            sumTree(root->right,sum);
        }
        // 当前节点加上右子树总和
        sum+=root->val;
        root->val = sum;
        // 左子树=父节点+父节点右子树总和
        if(root->left!=nullptr){
            sumTree(root->left,sum);
        }
    }
};