430.扁平化多级双向链表 (Medium)*
题目描述*
您将获得一个双向链表，除了下一个和前一个指针之外，它还有一个子指针，可能指向单独的双向链表。这些子列表可能有一个或多个自己的子项，依此类推，生成多级数据结构，如下面的示例所示。
扁平化列表，使所有结点出现在单级双链表中。您将获得列表第一级的头部。
示例*
示例:
输入:
1---2---3---4---5---6--nullptr
|
7---8---9---10--nullptr
|
11--12--nullptr
输出: 1-2-3-7-8-11-12-9-10-4-5-6-nullptr
代码*
class Solution {
public:
Node* flatten(Node* head) {
Node *cur = head;
while(cur != nullptr) {
if(cur->child != nullptr) {
Node *curNext = cur->next;
Node *ch = cur->child;
cur->next = ch;
ch->prev = cur;
cur->child = nullptr; // 当前节点的子指针要置空
while(ch != nullptr && ch->next != nullptr) {
ch = ch->next;
}
ch->next = curNext;
if(curNext != nullptr) {
curNext->prev = ch;
}
}
cur = cur->next;
}
return head;
}
};
最后更新: July 23, 2022