LeetCode430. 扁平化多级双向链表

最新推荐文章于 2024-10-17 16:33:41 发布

皮卡猴

最新推荐文章于 2024-10-17 16:33:41 发布

阅读量233

点赞数

分类专栏： leetcode刷题文章标签：链表数据结构 leetcode

本文链接：https://blog.csdn.net/flying_monkey_1/article/details/121498641

版权

leetcode刷题专栏收录该内容

252 篇文章 6 订阅

订阅专栏

这篇博客介绍了如何解决LeetCode第430题，即扁平化多级双向链表的问题。通过深度优先搜索（DFS）策略，将一个多级的双向链表转化为单级链表。代码中定义了Node结构体，并展示了如何遍历并连接各个节点，确保所有子节点被正确地连接到主链表中。

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

LeetCode430. 扁平化多级双向链表

1. 问题描述

在这里插入图片描述

2. 思路

3. 代码

/**
 * Definition for a Node.
 * type Node struct {
 *     Val int
 *     Prev *Node
 *     Next *Node
 *     Child *Node
 * }
 */

func flatten(root *Node) *Node {
    dfs(root)
    return root
}

func dfs(node *Node) *Node {
    var last *Node
    cur := node 
    for cur != nil {
        next := cur.Next
        // 处理子节点
        if cur.Child != nil {
            childLast := dfs(cur.Child)
            next = cur.Next
            
            // 将node与child连起来 
            cur.Next = cur.Child
            cur.Child.Prev = cur

            // 将last与next连起来
            if next != nil {
                childLast.Next = next
                next.Prev = childLast
            }

            // 将child置为空
            cur.Child = nil
            last = childLast
        } else {
            last = cur
        }
        cur = next
    }
    return last
}