剑指 Offer II 028. 展平多级双向链表

最新推荐文章于 2024-11-04 13:52:23 发布

daweq

最新推荐文章于 2024-11-04 13:52:23 发布

阅读量61

点赞数

文章标签：链表数据结构

本文链接：https://blog.csdn.net/daweq/article/details/130480696

版权

该文章描述了一种链表展平的方法，通过遍历每个节点，找到子链的最后一个节点，然后将子链插入到当前节点和其下一个节点之间，递归处理子链的子链，最终实现多级链表的展平。

摘要由CSDN通过智能技术生成

在这里插入图片描述
思路：
展平的规律是一个节点的子链展平后之后将插入该节点和它的下一个节点之间。也就是我们如果查到一个节点有子链，那么我们需要知道子链的最后一个节点，这样再把最后一个节点的next指向主链的next，再修改prev的值，就能实现插入子链的操作了。
因为子链还可能有子链，所以我们要用遍历的方式。

class Solution {
public:
    Node* flatten(Node* head)
    {
        flattenGetTail(head);
        return head;
    }

    Node* flattenGetTail(Node* head)
    {
        Node* node = head;
        Node* tail = nullptr;
        while (node != nullptr)
        {
            Node* next = node->next;
            if (node->child != nullptr)
            {
                Node* child = node->child;
                Node* childTail = flattenGetTail(node->child);

                node->child = nullptr;
                node->next = child;
                child->prev = node;
                childTail->next = next;
                if (next != nullptr)
                {
                    next->prev = childTail;
                }
                tail = childTail;
            }
            else
                tail = node;
            node = next;
        }
        return tail;
    }
};