430. 扁平化多级双向链表

最新推荐文章于 2022-05-14 07:57:32 发布

原创最新推荐文章于 2022-05-14 07:57:32 发布 · 116 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

121 篇文章

订阅专栏

本文介绍了一种算法，用于将包含子链表的多级双向链表扁平化为单级链表。通过先序遍历的方式，递归地处理每个节点及其子链表，最终实现所有节点在同一级别上。示例展示了输入和输出链表的结构变化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

您将获得一个双向链表，除了下一个和前一个指针之外，它还有一个子指针，可能指向单独的双向链表。这些子列表可能有一个或多个自己的子项，依此类推，生成多级数据结构，如下面的示例所示。

扁平化列表，使所有结点出现在单级双链表中。您将获得列表第一级的头部。

示例:

输入:
 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

以上示例的说明:

给出以下多级双向链表:

我们应该返回如下所示的扁平双向链表:

Review:

二叉树的先序遍历而已

Code:

class Solution {
    public Node flatten(Node head) {
        helper(head);
        return head;
    }

    private static Node helper(Node node){
        Node pre = node;
        while (node!=null){
            pre = node;
            if (node.child!=null){
                Node next = node.next;
                node.next = node.child;
                node.next.prev = node;
                node.child = null;
                Node last = helper(node.next);
                last.next = next;
                if (next!=null){
                    next.prev = last;
                }
            }
            node = pre.next;
        }
        return pre;
    }
}