[Leetcode]扁平化多级双向链表

题目

https://leetcode-cn.com/explore/learn/card/linked-list/197/conclusion/764/

代码

/*

// Definition for a Node.

class Node {

public:

    int val;

    Node* prev;

    Node* next;

    Node* child;

    Node() {}

    Node(int _val, Node* _prev, Node* _next, Node* _child) {

        val = _val;

        prev = _prev;

        next = _next;

        child = _child;

    }

};

*/

class Solution {

public:

    Node* flatten(Node* head) {

        if(head==nullptr)

            return nullptr;

        Node*ptr=head;

        while(ptr!=nullptr)

        {

            //当前结点有孩子结点

            if(ptr->child!=nullptr)

            {

                auto child=ptr->child;

                auto ptrNext=ptr->next;

                child->prev=ptr;

                ptr->next=child;

                ptr->child=nullptr;

                while(child->next!=nullptr)

                    child=child->next;

                //当前结点不是尾结点

                if(ptrNext!=nullptr)

                {

                    ptrNext->prev=child;

                    child->next=ptrNext;

                }

            }

                ptr=ptr->next;

        }

         return head;

    }

};

巴特西

[Leetcode]扁平化多级双向链表

题目

代码

最新文章

热门文章