您将获得一个双向链表,除了下一个和前一个指针之外,它还有一个子指针,可能指向单独的双向链表。这些子列表可能有一个或多个自己的子项,依此类推,生成多级数据结构,如下面的示例所示。
扁平化列表,使所有结点出现在单级双链表中。您将获得列表第一级的头部。
示例:
输入: 1---2---3---4---5---6--NULL | 7---8---9---10--NULL | 11--12--NULL 输出: 1-2-3-7-8-11-12-9-10-4-5-6-NULL
以上示例的说明:
给出以下多级双向链表:
我们应该返回如下所示的扁平双向链表:
/*
// Definition for a Node.
class Node {
public:
int val = NULL;
Node* prev = NULL;
Node* next = NULL;
Node* child = NULL;
Node() {}
Node(int _val, Node* _prev, Node* _next, Node* _child) {
val = _val;
prev = _prev;
next = _next;
child = _child;
}
};
*/
class Solution {
public:
Node* flatten(Node* head) {
helper(head, NULL);
return head;
}
Node* helper(Node* cur, Node* prev){
if(!cur) return NULL;
cur->prev = prev;
if(!cur->child){
if(cur->next) return helper(cur->next, cur);
return cur;
}
else{
Node* child = cur->child;
cur->child = NULL;
Node* next = cur->next;
Node* last = helper(child, cur);
cur->next = child;
if(next){
last->next = next;
return helper(next, last);
}
else{
return last;
}
}
}
};