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;
}
}