扁平化多级双向链表

扁平化多级双向链表

照抄官方题解，只是加些注释

本质上是将一个二叉树进行先序遍历，遍历时的操作是将当前节点与前驱连接起来，根节点没有前驱，所以给根节点加一个假头节点。

操作顺序为，

• 处理根节点
• 处理左子树
• 处理右子树

// 递归版
    public Node flatten(Node head) {
        if(head == null ) {
            return head;        //如果传入的是空表，不需要处理，直接返回
        }

        Node dummyHead = new Node(0,null,head,null);        //创建一个头结点，简化操作

        flattenDFS(dummyHead,head);         //递归的处理每个节点,第二个参数才是被处理的节点，第一个参数只是提供一个前驱供待处理节点连接

        dummyHead.next.prev = null;
        return dummyHead.next;

    }

    public Node flattenDFS(Node pre,Node cur){      //以cur作为当前处理的焦点

        if(cur == null){        //如果当前节点是null，什么也不做,把叶子节点返回，           递归出口
            return pre;
        }

        cur.prev = pre;         //步骤一：如果当前节点不是null，将当前节点与父节点连接          处理根节点
        pre.next = cur;

        Node temp = cur.next;           //temp暂存右子树

        Node tail = flattenDFS(cur, cur.child);     //步骤二：递归的处理左子树                 处理左子树

        cur.child = null;               //左子树处理完毕，不需要再保留child指针

        return flattenDFS(tail,temp);       //步骤三：  处理右子树， 并将处理结果连接在处理后的左子树上
    }

// 迭代版
    public Node flatten(Node head) {
        if(head == null){
            return head;
        }

        Node dummyHead = new Node(0,null,head,null);        //创建一个头结点，简化操作
        Node pre = dummyHead;
        Node cur;

        Deque<Node> stack = new ArrayDeque<>();
        stack.push(head);                   //将头节点入栈，使栈中至少有一个节点

        while (!stack.isEmpty()){           //当栈不为空时
            cur = stack.pop();
            pre.next = cur;                 //将当前处理的节点与其父节点连接。迭代法只处理当前节点，但是限制了子树的处理顺序
            cur.prev = pre;

            if (cur.next != null) stack.push(cur.next);       //如果当前节点有右子树，推入栈中

            if(cur.child != null){                          //如果当前节点有左子树，推入栈中
                stack.push(cur.child);
                cur.child = null;                       //顺便处理左子树引用
            }                                           //栈的先入后出保证了先处理左子树再处理右子树

            pre = cur;
        }

        dummyHead.next.prev = null;
        return dummyHead.next;

    }