题目描述
题干:
多级双向链表中,除了指向下一个节点和前一个节点指针之外,它还有一个子链表指针,可能指向单独的双向链表。
这些子列表也可能会有一个或多个自己的子项,依此类推,生成多级数据结构,如下面的示例所示。
给你位于列表第一级的头节点,请你扁平化列表,使所有结点出现在单级双链表中。
示例 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]
示例 3:
输入:head = []
输出:[]
题解思路
返回扁平化处理的双向链表,我们只需要把每个有 child 的节点的分支链表插入到他的母链表里即可
所以我们需要依次便利每个有 child 的节点,之后将分支链表的尾节点连接到 next
将当前的 next 连接到 child,最后将 child 置空。这样就完成了一次扁平化
正确代码
public Node flatten(Node head) {
dfs(head);
return head;
}
private Node dfs(Node node) {
Node cur = node;
Node last = null;
while (cur != null) {
Node next = cur.next;
// 如果有子节点,先遍历子节点
if (cur.child != null) {
Node childLast = dfs(cur.child);
next = cur.next;
// 将 node 与 child 相连
cur.next = cur.child;
cur.child.prev = cur;
// 如果 next 不为空,就将 last 与 next 相连
if (next != null) {
childLast.next = next;
next.prev = childLast;
}
// 将child置空
cur.child = null;
last = childLast;
} else {
last = cur;
}
cur = next;
}
return last;
}
class Node {
public int val;
public Node prev;
public Node next;
public Node child;
}
总结
双向链表和普通的链表相差不大,只需要注意插入节点的顺序即可,当然也可以采用递归解决
如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论,各自努力,你我最高处见