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]
解释：

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

扁平化后的链表如下图：

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flatten-a-multilevel-doubly-linked-list

分析

这题的提示是采用递归解法，我们也看到了，每遇到一个child就需要进去，并查看其还有没有child，如果有child需要继续进去连接。那么如果采用的是递归法，我们只需要做一次这样的连接child操作，即找出child，然后给出child的头节点，然后for循环找出child节点的尾结点，然后将其嵌入到当前节点。

注意: 当child节点被链接过后，一定要将当前节点的Child设置为nil，不然将其返回到上一层之后，它是会被再次读取的，然后就会造成无限循环，从而引发超时。

解法

/**
 * Definition for a Node.
 * type Node struct {
 *     Val int
 *     Prev *Node
 *     Next *Node
 *     Child *Node
 * }
 */

func flatten(root *Node) *Node {
    if root == nil {
        return nil
    }
    
    cur := root
    for cur != nil {
        if cur.Child != nil {
            newNext := flatten(cur.Child)
            // 找出child的尾结点
            tail := newNext 
            for tail.Next != nil {
                tail = tail.Next
            }
            // 将尾结点的设置为当前节点的下一个的Prev节点
            tail.Next = cur.Next
            if cur.Next != nil {
                cur.Next.Prev = tail
            }
            // 将当前节点next设置为child的头结点，将child的头结点Prev设置为当前节点
            cur.Next = newNext
            newNext.Prev = cur
            // 要注意这里要设置为nil，不然会超时！！！
            cur.Child = nil
        }
        cur = cur.Next
    }
    return root
}