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

430. 扁平化多级双向链表

写在前面:
最近事情比较多,马上要准备期末考试了,现在是在复习。然后又有数据库课设和计算机组成原理课设,好多事情要做,还有马上就要考六级笔试了,每天都要刷英语题,但是做算法题我是一定会坚持下去的,希望大家可以和我一起努力,谢谢大家。
原题链接:
https://leetcode.cn/problems/...leetcode 430.扁平化多级双向链表
题目描述:
你会得到一个双链表,其中包含的节点有一个下一个指针、一个前一个指针和一个额外的子指针。这个子指针可能指向一个单独的双向链表,也包含这些特殊的节点。这些子列表可以有一个或多个自己的子列表,以此类推,以生成如下面的示例所示的多层数据结构给定链表的头节点head,将链表扁平化,以便所有节点都出现在单层双链表中。让crr是一个带有子列表的节点。子列表中的节点应该出现在扁平化列表中的curr之后和curr.next之前。返回扁平列表的had。列表中的节点必须将其所有子指针设置为NULL。
最开始题目我觉得有点难理解,要看好几遍才可以理解,我建议你打开链接直接看leetcode里面的题目描述,我这里写的不是很清楚。
示例:
Leetcode 430. 扁平化多级双向链表_第1张图片
Leetcode 430. 扁平化多级双向链表_第2张图片

做题思路

深度优先遍历的思想,我这里用的栈来写深度优先遍历,没有用队列,遇到一个节点cur,if(cur->child==NULL)cur=dfs(cur->next); else{ 递归搜索到孩子节点为头节点的链表的尾部,将这几个点连接好(这里我用一张图来表示);}
Leetcode 430. 扁平化多级双向链表_第3张图片
注意:由于是双向链表,要修改连接的地方有四个。然后t1,t2分别是是孩子链表上的头和尾节点。

代码c++版

class Solution {
public:
    Node* dfs(Node* now){
        if(now==NULL) return NULL;
        Node* ans=NULL;
        if(now->child==NULL){
            if(now->next==NULL) return now;
            ans=dfs(now->next);
        }
        else{
            Node* t1=now->child;
            Node* t2=dfs(now->child);
            Node* t3=now->next;
            now->child=NULL;
            now->next=t1;
            t1->prev=now;
            t2->next=t3;
            if(t3==NULL) return t2;
            else t3->prev=t2;
            ans=dfs(t3);
        }
        return ans;
    }
    Node* flatten(Node* head) {
        dfs(head);
        return head;
    }
};

你可能感兴趣的:(Leetcode 430. 扁平化多级双向链表)