430. 扁平化多级双向链表

在这里插入图片描述
在这里插入图片描述

方法一:栈

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* prev;
    Node* next;
    Node* child;
};
*/
class Solution {
public:
    Node* flatten(Node* head) {
        if(!head)
            return NULL;
        Node *cur;
        stack <Node*> st;
        st.push(head);  //先将头结点放入栈中
        Node *pre = NULL;   //头结点没有前一个结点
        while(!st.empty()){
            cur=st.top();   //判断栈顶
            st.pop();
            if(cur->next)   //如果有next结点放入栈中,注意next先入栈
                st.push(cur->next);
            if(cur->child){     //如果有child结点放入栈中,注意child后入栈
                st.push(cur->child);
                cur->child = NULL;  //清空child指针
            }
            if(pre){    //与前一个结点连接,注意头结点没有前结点所以pre初始化为NULL
                pre->next=cur;
                cur->prev=pre;
            }
            pre=cur; //当前结点设为前结点
        }         
        return head;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值