[路飞]_leetcode-430-扁平化多级双向链表

567 阅读3分钟

「这是我参与2022首次更文挑战的第30天,活动详情查看:2022首次更文挑战

[题目地址] [B站地址]

多级双向链表中,除了指向下一个节点和前一个节点指针之外,它还有一个子链表指针,可能指向单独的双向链表。这些子列表也可能会有一个或多个自己的子项,依此类推,生成多级数据结构,如下面的示例所示。

给你位于列表第一级的头节点,请你扁平化列表,使所有结点出现在单级双链表中。

示例 1:

输入: head = [1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]
输出: [1,2,3,7,8,11,12,9,10,4,5,6]
解释:
输入的多级列表如下图所示:

扁平化后的链表如下图:

示例 2:

输入: head = [1,2,null,3]
输出: [1,3,2]
解释: 输入的多级列表如下图所示:

  1---2---NULL
  |
  3---NULL

示例 3:

输入: head = []
输出: []

如何表示测试用例中的多级链表?

以 示例 1 为例:

 1---2---3---4---5---6--NULL
         |
         7---8---9---10--NULL
             |
             11--12--NULL

序列化其中的每一级之后:

[1,2,3,4,5,6,null]
[7,8,9,10,null]
[11,12,null]

为了将每一级都序列化到一起,我们需要每一级中添加值为 null 的元素,以表示没有节点连接到上一级的上级节点。

[1,2,3,4,5,6,null]
[null,null,7,8,9,10,null]
[null,11,12,null]

合并所有序列化结果,并去除末尾的 null 。

[1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]

 

提示:

  • 节点数目不超过 1000
  • 1 <= Node.val <= 10^5

解题思路

本题要求我们把多级双向链表扁平化,也就是压成一级双向链表,所以我们需要把第一层每个节点的子链表接在它的后面,又因为子链表中的节点可能也会存在子链表,然后子链表中的节点的子链表中的节点也可能存在子链表,以此类推...,所以这必然是需要递归处理的一个问题。
这里我们以一层为例,如果当前节点有子链表,则要把子链表的头节点接在当前节点的后面,把子链表的尾节点接在当前节点的下一个节点的前面。
而对于子链表,我们采用相同的逻辑处理即可。

代码实现

var flatten = function (head) {
  let cur = head
  // 扫描当前链表
  while (cur) {
    // 如果当前节点有子链表
    if (cur.child) {
      // 获取下一个节点
      const next = cur.next,
        // 递归处理子链表并获取处理后的链表的头节点
        cHead = flatten(cur.child)
      // 将当前节点的子链表置空
      cur.child = null
      // 将处理后的子链表的头节点接在当前节点的后面
      ;(cur.next = cHead), (cHead.prev = cur)
      // cur 指针走到子链表的末尾节点
      while (cur.next) cur = cur.next
      // 将下一个节点接在子链表的末尾节点后面
      if (next) {
        cur.next = next
        next.prev = cur
      }
    }
    // 继续向后处理链表节点
    cur = cur.next
  }
  // 返回处理后的链表的头节点
  return head
}

至此我们就完成了 leetcode-430-扁平化多级双向链表

如有任何问题或建议,欢迎留言讨论!👏🏻👏🏻👏🏻