1. 题目
您将获得一个双向链表,除了下一个和前一个指针之外,它还有一个子指针,可能指向单独的双向链表。这些子列表可能有一个或多个自己的子项,依此类推,生成多级数据结构,如下面的示例所示。
扁平化列表,使所有结点出现在单级双链表中。您将获得列表第一级的头部。
输入:
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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. DFS
class Solution {
public:
Node* flatten(Node* head) {
Node *cur = head, *nextNode = NULL, *prevN = head;
//所有的指针记得初始化,不要以为后面要付值就不要初始值,怕后面if没进去
while(prevN)
{ //遍历链表
if(cur && cur->child) //当前有,且有子链
{
nextNode = cur->next;//记住上层cur后面的
cur->next = flatten(cur->child);//展平
cur->child->prev = cur;//子链头接上,展平
cur->child = NULL;//将cur孩子删掉
}
prevN = cur;//prevnode移动
if(cur)
cur = cur->next;//cur移动
if(!cur && nextNode)
{ //当cur到终点的下一个时,且上层后面还有nextnode
prevN->next = nextNode;//展平的终点prevN指向上层的nextnode
nextNode->prev = prevN;//反向接上
cur = prevN->next;//cur移动
nextNode = NULL;//已接好,暂时没有child的需要展平,nextNode清空
}
}
return head;
}
};