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到终点的下一个时,且上层后面还有nextnodeprevN->next = nextNode;//展平的终点prevN指向上层的nextnodenextNode->prev = prevN;//反向接上cur = prevN->next;//cur移动nextNode = NULL;//已接好,暂时没有child的需要展平,nextNode清空}}return head;}
};