0
点赞
收藏
分享

微信扫一扫

LeetCode练习43:展平多级双向链表

您好 2022-04-22 阅读 69

LeetCode链接:力扣

题目:

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

给定位于列表第一级的头节点,请扁平化列表,即将这样的多级双向链表展平成普通的双向链表,使所有结点出现在单级双链表中。

示例:

输入: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]
解释:

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

扁平化后的链表如下图:

代码:

/*
// Definition for a Node.
class Node {
    public int val;
    public Node prev;
    public Node next;
    public Node child;
};
*/

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

    public Node dfs(Node head){
        Node curr = head;
        Node last = null;       //最后一个节点

        while(curr != null){
            Node next = curr.next;
            
            if(curr.child != null){
                Node childLast = dfs(curr.child);
                next = curr.next;

                //node与chile相连
                curr.next = curr.child;
                curr.child.prev = curr;

                //如果next不为空,则next与childLast相连
                if(next != null){
                    childLast.next = next;
                    next.prev = childLast;
                }

                //将child置空
                curr.child = null;
                last = childLast;       //将child末尾设为最后节点

            }else{
                last = curr;
            }
            curr = next;
        }
        return last;
    }
}
举报

相关推荐

0 条评论