114 二叉树展开为链表

114 二叉树展开为链表

题目：给你二叉树的根结点 root ，请你将它展开为一个单链表：展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。展开后的单链表应该与二叉树 先序遍历 顺序相同。

示例：

思考；

• 前序遍历
• 生成节点，左孩子为空，右孩子指向下一个结点

public class Solution {
    /**
     * 二叉树展开为链表
     * 输入：root = [1,2,5,3,4,null,6]
     * 输出：[1,null,2,null,3,null,4,null,5,null,6]
     * 注意：
     * （1）right 子指针指向链表中下一个结点，而左子指针始终为 null 。
     * （2）展开后的单链表与二叉树先序顺序相同
     */
    public void flatten(TreeNode root) {
        if (root == null) {
            return;
        }
        LinkedList<TreeNode> stack = new LinkedList<>();
        stack.push(root);
        List<TreeNode> list = new ArrayList<>();

        // 结果集中的单链表是前序遍历的结果
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();

            list.add(node);
            // 非递归前序：先右后左
            if (node.right != null) {
                stack.push(node.right);
            }
            if (node.left != null) {
                stack.push(node.left);
            }
        }

        for (int i = 0; i < list.size() - 1; i++) {
            TreeNode pre = list.get(i);
            TreeNode next = list.get(i + 1);
            // 单链表结点左孩子为空，右孩子指向下一个结点
            pre.left = null;
            pre.right = next;
        }

    }
}