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

题目：

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。

示例：

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

思路：

对于二叉搜索树，正常的中序遍历得到的是一个升序序列，变为累加树就是依次遍历节点，将当前节点后面的全部元素值相加，相加的结果赋值给当前节点，但是这样做时间复杂度太高了。

所以考虑逆序中序遍历，右根左的顺序去遍历，这样右边遍历到的都是大数，然后再遍历一个后边的就累加上前面遍历过的大于当前值的节点，并将累加后的结果给到当前节点，直到遍历完整个二叉树，构成累加树。

这里用栈实现二叉树的逆序中序遍历，维护累加值，给到当前遍历的节点。

记得前面保留原来的根节点，最后返回根节点。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def convertBST(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        stack = []
        sum = 0
        newroot = root
        while stack or root:
            while root:
                stack.append(root)
                root = root.right
            root = stack.pop()
            sum += root.val
            root.val = sum
            root = root.left
        return newroot