LeetCode刷题实战430:扁平化多级双向链表
共 2135字,需浏览 5分钟
·
2021-11-09 14:52
示例
解题
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;
}
};