深搜+栈==实现扁平化多级双向链表

Posted 2021-10-14 C_YCBX Py_YYDS

文章目录

• • 题目
  • 解题概述
  • 解题代码

题目

OJ平台

解题概述

看了这个扁平化实现的简介，第一时间就能想到用深搜，但注意还要实现对尾部连接，这个我是想到用栈存储然后再处理的方式，官方题解就是一次递归就解决了。

解题代码

//一看就老深搜了
class Solution {
public:
    stack<Node*>St; //用栈记录下没有被处理的结点
    Node* rear = nullptr; //记录尾指针
    Node* flatten(Node* head) {
        if(!head)return nullptr;
        Node* res = dfs(head);
        handle_r();
        return res;
    }
    Node* dfs(Node* head){
        if(head->next==nullptr&&head->child==nullptr){//无法往下走的情况
            rear = head;
        }
        else if(head->child){//有下一层级的情况
            if(head->next)
            St.push(head->next);
            Node* child = head->child;
            child->prev = head;
            head->child = nullptr;
            head->next = dfs(child);
        }else {//层级为1的情况
            head->next = dfs(head->next);
        }
        return head;
    }
    //根据记录的尾指针进行连接
    void handle_r(){
        Node* r = rear;
        while(!St.empty()){
            Node* p = St.top();St.pop();
            p->prev = r;r->next = p;
            if(St.empty())
                break;
            while(p->next){
                p = p->next;
            }
            r = p;
        }
    }
};