将二叉搜索树改为累加树

package main
import _"fmt"
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return TreeNode类
*/

var sum int
func convertBST( root *TreeNode ) *TreeNode {
    arr:=order(root)
    for i,x:=range arr{
        ori:=x.Val
        arr[i].Val=sum
        sum-=ori
    }
    return root
}

func order(root *TreeNode)[]*TreeNode{
    if root==nil{
        return nil
    }
    ans:=[]*TreeNode{}
    sum+=root.Val
    ans=append(ans,order(root.Left)...)
    ans=append(ans,root)
    ans=append(ans,order(root.Right)...)
    return ans
}