Leetcode 430：扁平化多级双向链表

题目描述

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

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

示例:

输入:
 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
 

以上示例的说明:

给出以下多级双向链表:

 

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flatten-a-multilevel-doubly-linked-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

class Solution {
public:
    Node* flatten(Node* head) {
        Node *cur = head;
        while(cur != NULL){
            if(cur->child != NULL){
                Node* post = cur->next;
                Node* child = cur->child;
                cur->child = NULL;
                cur->next = child;
                child->prev = cur;
                child = flatten(child);
                while(child->next != NULL) child=child->next;
                child->next = post;
                if(post != NULL) post->prev = child;
            }
            cur = cur->next;
        }
        return head;
    }
};