Skip to content
本页目录

0430-扁平化多级双向链表

https://leetcode.cn/problems/flatten-a-multilevel-doubly-linked-list

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

给你位于列表第一级的头节点,请你扁平化列表,使所有结点出现在单级双链表中。

示例 1:

输入:head = [1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]
输出:[1,2,3,7,8,11,12,9,10,4,5,6]
解释:

输入的多级列表如下图所示:

扁平化后的链表如下图:

示例 2:

输入:head = [1,2,null,3]
输出:[1,3,2]
解释:

输入的多级列表如下图所示:

  1---2---NULL
  |
  3---NULL
示例 3:

输入:head = []
输出:[]

如何表示测试用例中的多级链表?

以 示例 1 为例:

1---2---3---4---5---6--NULL | 7---8---9---10--NULL | 11--12--NULL 序列化其中的每一级之后:

[1,2,3,4,5,6,null] [7,8,9,10,null] [11,12,null] 为了将每一级都序列化到一起,我们需要每一级中添加值为 null 的元素,以表示没有节点连接到上一级的上级节点。

[1,2,3,4,5,6,null] [null,null,7,8,9,10,null] [null,11,12,null] 合并所有序列化结果,并去除末尾的 null 。

[1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]

提示:

节点数目不超过 1000 1 <= Node.val <= 10^5

思路

扁平化节点的时候,遇到子节点的之后,就开始执行扁平化操作,然后将返回的节点插入到当前节点和后一个节点中去

参考代码

csharp
//正确代码
public class Solution {
    public Node Flatten(Node head) {
        Node cur = head;
        while(cur != null){
            if(cur.child != null){
                break;
            }
            cur = cur.next;
        }
        if(cur != null){ //说明找到 child 节点
        	//取出后面节点
        	Node next = cur.next;
        	//扁平化子节点
        	Node flatten = Flatten(cur.child);
            //设置Child为空(必须)
            cur.child = null;
        	//将子节点接入
        	cur.next = flatten;
            flatten.prev = cur;
        	while(cur.next != null){
        		cur = cur.next;
        	}
        	//将前面的后续节点接入到最后
        	cur.next = next;
            if(next != null){
                next.prev = cur;
            }
        }
        return head;
    }
}






//调试代码参考
public class Solution {
    public Node Flatten(Node head) {
        Node cur = head;
        while(cur != null){
            //Console.WriteLine("Loop {0}",cur.val);
            if(cur.child != null){
                //Console.WriteLine("Found Child {0}",cur.child.val);
                break;
            }
            cur = cur.next;
        }
        if(cur != null){ //说明找到 child 节点
            //Console.WriteLine("Find Child {0}",cur.child.val);
        	//取出后面节点
        	Node next = cur.next;
            //Console.WriteLine("Next {0}",next.val);
        	//扁平化子节点
        	Node flatten = Flatten(cur.child);
            //设置Child为空(必须)
            cur.child = null;
        	//将子节点接入
        	cur.next = flatten;
            flatten.prev = cur;
            //Console.WriteLine("Flatten {0}, prev{1}",flatten.val,flatten.prev.val);
        	while(cur.next != null){
        		cur = cur.next;
        	}
        	//将前面的后续节点接入到最后
            //Console.WriteLine("Latest CUR {0}, next {1}",cur.val,next.val);
        	cur.next = next;
            if(next != null){
                next.prev = cur;
            }
            //Console.WriteLine("{0} prev is {1}",next.val, next.prev.val);
        }

        //Console.WriteLine("return {0}",head.val);
        //测试代码:重新输出链表
        cur = head;
        Console.WriteLine("重新输出链表");
        if(cur != null){
            Console.WriteLine("Output node {0}",cur.val);
        }
        while(cur.next != null){
            cur = cur.next;
            Console.WriteLine("Output node {0}",cur.val);
        }
        if(cur != null){
            Console.WriteLine("Output pre node {0}",cur.val);
        }
        while(cur.prev != null){
            cur = cur.prev;
            Console.WriteLine("Output pre node {0}",cur.val);
        }
        Console.WriteLine("重新输出链表结束");
        
        return head;
    }
}

复习:20220519

csharp
public class Solution {
    public Node Flatten(Node head) {
        if(head == null){
            return null;
        }
        Node cur = head;
        while(cur != null){
            if(cur.child != null){
                Node child = Flatten(cur.child);
                cur.child = null;
                //缓存 next指针
                Node next = cur.next;
                //指向 next 为child
                cur.next = child;
                child.prev = cur;
                //child move到最后
                while(child.next != null){
                    child = child.next;
                }
                //child接上当前next
                child.next = next;
                if(next != null){
                    next.prev = child;
                }
            }
            cur = cur.next;
        }
        return head;
    }
}

Released under the MIT License.