LeetCode 430. 扁平化多级双向链表（DFS）

1. 题目

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

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

输入:
 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. DFS

class Solution {
public:
    Node* flatten(Node* head) {
        Node *cur = head, *nextNode = NULL, *prevN = head;
        //所有的指针记得初始化，不要以为后面要付值就不要初始值，怕后面if没进去
        while(prevN)
        {	//遍历链表
            if(cur && cur->child) //当前有，且有子链
            {
                nextNode = cur->next;//记住上层cur后面的
                cur->next = flatten(cur->child);//展平
                cur->child->prev = cur;//子链头接上，展平
                cur->child = NULL;//将cur孩子删掉
            }
            prevN = cur;//prevnode移动
            if(cur)
                cur = cur->next;//cur移动
            if(!cur && nextNode)
            {	//当cur到终点的下一个时，且上层后面还有nextnode
                prevN->next = nextNode;//展平的终点prevN指向上层的nextnode
                nextNode->prev = prevN;//反向接上
                cur = prevN->next;//cur移动
                nextNode = NULL;//已接好，暂时没有child的需要展平，nextNode清空
            }
        }
        return head;
    }
};