题目描述
多级双向链表中，除了指向下一个节点和前一个节点指针之外，它还有一个子链表指针，可能指向单独的双向链表。这些子列表也可能会有一个或多个自己的子项，依此类推，生成多级数据结构，如下面的示例所示。 给你位于列表第一级的头节点，请你扁平化列表，使所有结点出现在单级双链表中。
Example 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] 解释： 输入的多级列表如下图所示：
扁平化后的链表如下图：
Example 2:
输入：head = [1,2,null,3] 输出：[1,3,2] 解释： 输入的多级列表如下图所示： 1—2—NULL | 3—NULL
Example 3:
输入：head = [] 输出：[]
限制
- 节点数目不超过 1000
- 1 <= Node.val <= 10^5
解题思路
- dfs扩展
C++代码
/*
// Definition for a Node.
class Node {
public:
int val;
Node* prev;
Node* next;
Node* child;
};
*/
class Solution {
public:
Node* dfs(Node* head){
if(head == nullptr || (head -> next == nullptr && head -> child == nullptr)){
return head;
}
Node* nodeNext = nullptr;
Node* lastNode = nullptr;
while(head){
if(head -> child != nullptr){
nodeNext = head -> next;
head -> next = head -> child;
head -> child -> prev = head;
Node* nodeChild = head -> child;
head -> child = nullptr; // 递归的结束入口
head = dfs(nodeChild); // 递归
head -> next = nodeNext;
if(nodeNext){
nodeNext -> prev = head;
}
}
lastNode = head;
head = head -> next;
}
return lastNode;
}
Node* flatten(Node* head) {
dfs(head);
return head;
}
};
