leetcode(538). Convert BST to Greater Tree

question

Given a Binary Search Tree (BST), convert it to a Greater Tree such
that every key of the original BST is changed to the original key plus
sum of all keys greater than the original key in BST.

solution

把原有二叉排序树转化为一棵新的二叉树，使得这棵树上的节点的值是原对应节点的值加上所有大于它的值的和。

因为原来的树是一颗二叉排序树，所以可以知道大于某个节点的所有节点在且只在它的右子树上，因此我们可以用右子树-根-左子树的顺序遍历二叉树，同时使用全局变量记录经过的节点的和。

全局变量用法，首先要在全局作用域声明一个变量，然后如果在函数、匿名函数或类中修改变量则需要使用global声明使用的是一个全局变量，如果只是使用而不修改的话则无需使用global声明。更多参考python变量作用域

x = 0
def tra(root):
    global x
    if not root:
        return

    tra(root.right)
    tmp = root.val
    root.val += x
    # print(root.val, x)
    x += tmp

    tra(root.left)


class Solution(object):
    def convertBST(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        global x
        x = 0
        tra(root)

        return root

leetcode上不支持py3，所以无法使用nonlocal关键字，

class Solution(object):
    def convertBST(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        x = 0
        def tra(root):
            nonlocal x
            if not root:
                return

            tra(root.right)
            tmp = root.val
            root.val += x
            # print(root.val, x)
            x += tmp

            tra(root.left)
        tra(root)

        return root

在python2上可以使用实例变量模拟闭包外变量。

class Solution(object):
    def convertBST(self, root):
        self.tempsum = 0
        def decorder(root): 
            if not root: 
                return
            decorder(root.right)
            root.val += self.tempsum
            self.tempsum = root.val
            decorder(root.left)
        decorder(root)
        return root