【LeetCode】链表——扁平化多级双向链表

最新推荐文章于 2025-05-21 00:45:00 发布

原创最新推荐文章于 2025-05-21 00:45:00 发布 · 222 阅读

0 ·

CC 4.0 BY-SA版权

C++ 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种解决LeetCode上多级双向链表展平问题的方法，通过遍历链表并调整child节点，将多级链表转换为单级链表。文章提供了详细的算法思路和C++代码实现。

链接：https://leetcode-cn.com/explore/learn/card/linked-list/197/conclusion/764/
在这里插入图片描述
思路：顺着child结点链接链表前部分，将child后面的部分按顺序链起来，遍历完原链表之后，再将两部分整合。
代码：

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* prev;
    Node* next;
    Node* child;
};
*/
class Solution {
public:
    Node* flatten(Node* head) {
        if(head==NULL) return NULL;
        Node *p=head;
        Node *l=NULL;
        while(p)
        {
            if(p->child)
            { 
                Node *n=p;
                while(n->next)
                {
                    n=n->next;     
                }               
                if(l)
                {      
                    n->next=l;
                    l->prev=n;
                }
                if(p->next)
                {
                    l=p->next;
                    l->prev=NULL;
                }
                                        
                p->next=p->child;
                p->child->prev=p;
                p->child=NULL;
            }
            if(p->next) p=p->next;
            else break;
        }
        p->next=l;
        if(l) {l->prev=p;}
        return head;
    }
};