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

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

---

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

例如：

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

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

解题思路：对二叉搜索树采用右中左的顺序遍历树节点，得到的值就是从大到小排列的。使用一个变量记录节点值的累计和，并将该变量记录的累计和不断赋予给对应的节点。

Python3代码如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def convertBST(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        self.sum_node = 0
        def inorder(node):
            if not node:
                return None
            inorder(node.right)
            self.sum_node += node.val
            node.val = self.sum_node
            inorder(node.left)
        inorder(root)
        return root