剑指 Offer II 28. 展平多级双向链表

题目

• 链接 剑指 Offer II 28. 展平多级双向链表
• 描述

多级双向链表中，除了指向下一个节点和前一个节点指针之外，它还有一个子链表指针，可能指向单独的双向链表。这些子列表也可能会有一个或多个自己的子项，依此类推，生成多级数据结构，如下面的示例所示.给定位于列表第一级的头节点，请扁平化列表，即将这样的多级双向链表展平成普通的双向链表，使所有结点出现在单级双链表中。

• 输入与输出

输入：head = [1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]
输出：[1,2,3,7,8,11,12,9,10,4,5,6]

示例
把 child 看成是整体插入到父节点的 next 中

思路

可以使用 栈 来操作。思路的话就是利用 栈 的特性，

1. 如果 next 节点不为 null，添加 next 节点；
2. 再判断 child 节点是否为 null，不为 null 则添加。

代码

• 使用 栈

/*
// Definition for a Node.
class Node {
    public int val;
    public Node prev;
    public Node next;
    public Node child;
};
*/

class Solution {
    public Node flatten(Node head) {
        if (head == null) {
            return null;
        }
        
        Node newHead = new Node(-1,null,null,null);
        Node prev = newHead;
        Stack<Node> stack = new Stack<>();
        stack.push(head);
        
        while (!stack.isEmpty()) {
        	// 将当前节点的关系置为合理的关系
            Node cur = stack.pop();
            prev.next = cur;
            cur.prev = prev;
            prev = cur;
            
            if (cur.next != null) {
                stack.push(cur.next);
            }
            if (cur.child != null) {
                stack.push(cur.child);
                // 将 child 节点置为 null
                cur.child = null;
            }
        }
        
        // 将真正节点的 prev 置为 null
        // 返回 prev 节点的下一个节点, 也就是头节点
        newHead.next.prev = null;
        return newHead.next;
    }
}