538. 把二叉搜索树转换为累加树

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

例如:

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

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

中序遍历倒过来搞起!

class Solution {
    int pre = 0;
    public void dfs(TreeNode root){
        if(root != null){
            dfs(root.right);
            root.val = pre + root.val;
            pre = root.val;
            dfs(root.left);
        }            
    }
    public TreeNode convertBST(TreeNode root) {
        dfs(root);
        return root;
    }
}

猜你喜欢

转载自blog.csdn.net/xuchonghao/article/details/80770611