LeetCode刷题【430】扁平化多级双向链表

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

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

 

示例 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

今天每日一题，直接上代码

class Solution {
    Node preNode;
    public Node flatten(Node head) {
        if (null==head){
            return null;
        }
        preNode = head;
        Node next = head.next;
        dfs(head.child);
        head.child = null;
        dfs(next);
        return head;
    }

    public void dfs(Node node){
        if (null == node){
            return;
        }
        preNode.next = node;
        node.prev = preNode;
        preNode = node;
        Node next = node.next;
        dfs(node.child);
        node.child = null;
        dfs(next);
    }
}

其实就深搜。然后按照前序遍历的顺序重新把指针指向，代码里我重新定义了一个dfs方法，而原来的flatten方法中的逻辑也是一样的，唯一的区别就是flatten方法中没有做将head结点和preNode结点重新连接的操作，因为在头结点之前已经没有额外的一个preNode了。

这里没有在单独的dfs方法中为了第一个头结点的preNode做一次额外的判断，所以就重新写到了flatten中，避免在之后每个节点的递归过程中都要再做一次判断。

至于每次递归的逻辑

1. 把preNode的next指向到当前节点，当前节点的prev指向到preNode节点
2. preNode更新为当前节点
3. 额外的取到当前节点的next节点，暂时存起来
4. 递归处理当前节点的child节点，从步骤1开始重新执行，执行完毕后对应节点的next节点会被更新成新的其他的节点，所以原来的next节点要在3步骤的时候存起来。
5. 把当前节点的child节点的指向关系置为null。child分支部分的内容已经处理完毕，可以清空了
6. 继续递归处理next分支的信息，注意这里是处理的之前3步骤存下来的next节点，不是当前节点的next节点，因为当前节点的next节点已经在步骤4中被更新了，如果还是处理的当前节点的next节点，这个链表会变成环