[二叉树遍历|BST]leetcode 538 把二叉搜索树转换为累加树

[二叉树遍历|BST]leetcode 538 把二叉搜索树转换为累加树

1.题目

题目链接
给定一个二叉搜索树（Binary Search Tree），把它转换成为累加树（Greater Tree)，使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如：

输入: 原始二叉搜索树:
              5
            /   \
           2     13

输出: 转换为累加树:
             18
            /   \
          20     13

2.分析

2.1.要干什么

使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
那么现在的问题就是怎么找到“所有比当前结点值更大的结点”。

2.2.一个性质

对于二叉搜索树而言，进行中序遍历，可以得到一个递增序列。
例如，对

   5
 /   \
2     13

进行中序遍历，得到的序列是2,5,13。

2.3.不同点

对于当前遍历到的结点，我们需要将其加上“比其值更大的结点”的值，因此实际上应该先访问值更大的结点，求的是一个降序序列，也就是要将原有中序遍历的顺序给反过来。即：

TreeNode* convertBST(TreeNode* root) {
    if(root != nullptr) {
        convertBST(root->right);  //遍历右子树
		//do something...         //遍历当前结点，执行累加过程
        convertBST(root->left);   //遍历左子树
    }
    return root;
}

而累加的过程可以通过设置一个全局变量来完成。

3.代码

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