【LeetCode刷题（简单程度）】538. 把二叉搜索树转换为累加树

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

例如：

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：看到二叉搜索树应该想到中序遍历，因为常规中序遍历的BST是从小到大排序的，所以这题我们观察可以发现，比一个节点大的都是靠右边的值，所以我们可以按照 右 中 左的情况排序，这样就可以将大的数排在前面，利用一个变量记录累加值就行了。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int num = 0;
    TreeNode* convertBST(TreeNode* root) {  
        TreeNode* cur = root;
        inOrder(root);
        return cur;
    }
    void inOrder(TreeNode* root)//中序遍历
    {
        if(!root)
            return;
        //遍历右子树
        inOrder(root->right);
        //遍历顶点 注意此时累加
        root->val += num;
        num = root->val;
        //遍历左子树
        inOrder(root->left);
    }
};

思路二：非递归版本

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int num = 0;
    TreeNode* convertBST(TreeNode* root) {  
        TreeNode* cur = root;
        inOrder(root);
        return cur;
    }
    void inOrder(TreeNode* root)
    {
        stack<TreeNode*> s;
        TreeNode* p = root;
        while(p!=NULL||!s.empty())
        {
            while(p!=NULL)
            {
                s.push(p);
                p = p->right;//先达到最右节点
            }
            if(!s.empty())
            {
                p = s.top();
                s.pop();
                p->val += num;//更新节点为累加值
                num = p->val;//更新num
                p = p->left;
            }
        }
    }
};