二叉树展开为单链表

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

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
*/
func expandTree( root *TreeNode )  {
    arr:=[]*TreeNode{}
    var order func(*TreeNode)
    order=func(root *TreeNode){
        if root==nil{
            return
        }
        arr=append(arr,root)
        order(root.Left)
        order(root.Right)
    }
    order(root)
    for i,p:=range arr{
        p.Left=nil
        if i<len(arr)-1{
            p.Right=arr[i+1]
        }else{
            p.Right=nil
        }
    }
}