力扣——二叉树展开为链表

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

例如，给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为：

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
//深度优先遍历
//思想：先序遍历树，将节点插入到list中，再构造链表
//注意：遍历返回之前要将节点的左右子节点为null
class Solution {
    public void flatten(TreeNode root) {
        List<TreeNode> list = new ArrayList<>();
        if(root == null) return;
        fun(root,list);
        for(int i=0;i<list.size()-1;i++){
            list.get(i).right=list.get(i+1);
        }
    }
    
    public void fun(TreeNode root,List<TreeNode> list){
        list.add(root);
        if(root.left!=null)
            fun(root.left,list);
        if(root.right!=null)
            fun(root.right,list);
        root.left=root.right=null;
    }
}