【leetcode430】扁平化多级双向链表 Java题解

leetcode分类下所有的题解均为作者本人经过权衡后挑选出的题解，在易读和可维护性上有优势

每题只有一个答案，避免掉了太繁琐的以及不实用的方案，所以不一定是最优解

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

扁平化列表，使所有结点出现在单级双链表中。您将获得列表第一级的头部。

 

示例:

输入:
 1---2---3---4---5---6--NULL
         |
         7---8---9---10--NULL
             |
             11--12--NULL

输出:
1-2-3-7-8-11-12-9-10-4-5-6-NULL

 

以上示例的说明:

给出以下多级双向链表:

我们应该返回如下所示的扁平双向链表:

/*
// Definition for a Node.
class Node {
    public int val;
    public Node prev;
    public Node next;
    public Node child;

    public Node() {}

    public Node(int _val,Node _prev,Node _next,Node _child) {
        val = _val;
        prev = _prev;
        next = _next;
        child = _child;
    }
};
*/
class Solution {
    public Node flatten(Node head) {
        Node cur = head;
        while(cur != null){
            if(cur.child == null){
                cur = cur.next;
                continue;
            }
            Node child = cur.child;
            Node childTail = child;
            cur.child = null;
            child.prev = cur;
            
            while(childTail.next != null)
                childTail = childTail.next;
            childTail.next = cur.next;
            if(cur.next != null)
                cur.next.prev = childTail;
            cur.next = child;
            cur = cur.next;
        }
        return head;
    }
}

 思路：

• 先将10 给 4，再将4.next.pre 指向 10