114. 二叉树展开为链表 golang

114. 二叉树展开为链表

给定一个二叉树，原地将它展开为链表。

例如，给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为：

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

Code

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func flatten(root *TreeNode)  {
    if root == nil {
        return 
    }
    flatten(root.Left)
    flatten(root.Right)

    temp := root.Right

    root.Right, root.Left = root.Left, nil

    for root.Right != nil {
        root = root.Right
    }

    root.Right = temp 
}