【每日一题】430. 扁平化多级双向链表

多级双向链表中，除了指向下一个节点和前一个节点指针之外，它还有一个子链表指针，可能指向单独的双向链表。这些子列表也可能会有一个或多个自己的子项，依此类推，生成多级数据结构，如下面的示例所示。

给你位于列表第一级的头节点，请你扁平化列表，使所有结点出现在单级双链表中。

示例 1：
输入：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]
解释：

输入的多级列表如下图所示：

扁平化后的链表如下图：

思路：前序遍历即可，这实际上就是 一棵二叉树。

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

class Solution {
public:
    Node* flatten(Node* head) {
        if (head == nullptr) 
            return nullptr;
        stack<Node*> s;
        Node* cur = head;

        while (true) {
            if (cur->child != nullptr) {
                //将next节点入栈
                if (cur->next != nullptr) {
                    s.push(cur->next);
                }

                //将子链表进行扁平化操作
                cur->next = cur->child;
                cur->child->prev = cur;
                cur->child = nullptr;
            }

            //遍历子链表的下一个节点或是从栈中弹出next节点
            if (cur->next != nullptr) {
                cur = cur->next;
            } else if(!s.empty()) {
                Node* next = s.top();
                s.pop();
                cur->next = next;
                next->prev = cur;
                cur = next;
            } else {
                return head;
            }
        }
    }
};