问题描述:
您将获得一个双向链表,除了下一个和前一个指针之外,它还有一个子指针,可能指向单独的双向链表。这些子列表可能有一个或多个自己的子项,依此类推,生成多级数据结构,如下面的示例所示。
扁平化列表,使所有结点出现在单级双链表中。您将获得列表第一级的头部。
输入:
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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flatten-a-multilevel-doubly-linked-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
执行结果:
代码描述:
class Solution {
public:
Node* flatten(Node* head) {
if(head == NULL)
return NULL;
for(Node *m = head; m != NULL; m = m->next)
{
Node *t = m->next;
if(m->child)
{
Node* n = m->child;
m->next = n;
n->prev = m;
m->child = NULL; // 注意此处child指针赋值为NULL
while(n && n->next)
{
n = n->next;
}
n->next = t;
if(t != NULL)
t->prev = n;
}
}
return head;
}
};
思路如下: