Leetcode538. 把二叉搜索树转换为累加树c++实现

Leetcode538. 把二叉搜索树转换为累加树c++实现

题目描述

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

分析

首先，二叉搜索树按照中序遍历的序列是单调的，这里用一个栈来存储中序遍历的结果，边遍历边累加。

代码实现

  /**
     * 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:
        TreeNode* convertBST(TreeNode* root) {
            stack<TreeNode*> mystack; 
            TreeNode* cur = root; 
            int add = 0; 
            while(cur != NULL || !mystack.empty())
            {
                if(cur != NULL)
                {
                    mystack.push(cur);
                    cur = cur->right;
                }
                else 
                {
                    cur = mystack.top();
                    mystack.pop();
                    cur->val += add;
                    add = cur->val;
                    cur = cur->left;
                }
            }
            return root; 
        } 
    };