题目链接

剑指 Offer II 028. 展平多级双向链表 mid

题目描述

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

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

示例 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 = []
输出：[]

提示：

• 节点数目不超过 1000
• 1<=Node.val<=1051 <= Node.val <= 10^51<=Node.val<=105

解法 ： dfs

类似于 树的前序遍历。

我们用 pre来记录当前结点 root的前驱结点，用 nextNode来记录当前结点 root的后继结点。

如果 pre不为空，就让前驱结点pre的后继指针指向 root，即 pre.next = root。

当前结点root的前驱指针直接指向 pre，即 root.prev = pre。

接着再让 pre变为当前结点，先递归处理当前结点的子结点 root.child（如果 root.child != null的情况下），处理完毕之后，将root.child置为空，即 root.child = null表明其处理完毕，接着再递归处理下一个结点 nextNode。

时间复杂度： O(n)O(n)O(n) nnn是结点的数量

C++代码：

/*
// Definition for a Node.
class Node {
public:int val;Node* prev;Node* next;Node* child;
};
*/class Solution {
public:Node * pre = nullptr;void dfs(Node * root){if(root == nullptr) return;if(pre != nullptr) pre->next = root;Node * nextNode = root->next;root->prev = pre;pre = root;dfs(root->child);root->child = nullptr;dfs(nextNode);}Node* flatten(Node* head) {dfs(head);return head;}
};

Python代码：

"""
# Definition for a Node.
class Node:def __init__(self, val, prev, next, child):self.val = valself.prev = prevself.next = nextself.child = child
"""class Solution:      def flatten(self, head: 'Node') -> 'Node':pre = Nonedef dfs(root : 'Node') -> 'void':if root == None:returnnonlocal preif pre != None:pre.next = rootroot.prev = prenextNode = root.nextpre = rootdfs(root.child)root.child = Nonedfs(nextNode)dfs(head)return head