Leetcode 430.扁平化多级双向链表 c语言
程序员文章站
2022-04-24 16:46:17
...
Leetcode 430.扁平化多级双向链表 c语言
题目描述:
您将获得一个双向链表,除了下一个和前一个指针之外,它还有一个子指针,可能指向单独的双向链表。这些子列表可能有一个或多个自己的子项,依此类推,生成多级数据结构,如下面的示例所示。
扁平化列表,使所有结点出现在单级双链表中。您将获得列表第一级的头部。
示例:
输入:
1---2---3---4---5---6---NULL
|
7---8---9---NULL
|
11--12--NULL
输出:
1-2-3-7-8-11-12-9-10-4-5-6-NULL
以上示例的说明:
给出以下多级双向链表:
我们应该返回下图所示的扁平双向链表:
解题方法:
我们可以采用逐级扁平化的方法。使一个指针P标记第一个含有子链的节点,然后另一个指针L寻找P的子链的末节点,将P的后一个节点连接到L节点的后面。此时第一层扁平化结束,接着让P寻找下一个含有子链的节点,以此类推…直到P指向空。
代码如下:
class Solution {
public:
Node* flatten(Node* head) {
Node *p, *l = NULL;
p = head;
while (p != NULL)
{
if (p->child != NULL)
{
l = p->child;
l->prev = p;
while (l->next != NULL)
{
l = l->next;
}
l->next = p->next
if(l->next!=NULL)
l->next->prev = l;
p->next = p->child;
p->child = NULL;
}
p = p->next;
}
return head;
}
};
上一篇: Hibernate 【第一章 基础】