LeetCode 538. Convert BST to Greater Tree

538. Convert BST to Greater Tree（把二叉搜索树转换为累加树）

链接

https://leetcode-cn.com/problems/convert-bst-to-greater-tree

题目

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

例如：

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

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

注意：本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/ 相同

思路

直接递归，新建一个sum拿来存储值，然后修改结点值，这里记得右中左的方式。

代码

  int sum = 0;

  public TreeNode convertBST(TreeNode root) {
    if (root != null) {

      convertBST(root.right);
      sum += root.val;
      root.val = sum;
      convertBST(root.left);
    }
    return root;
  }