LeetCode偶尔一题 —— 430. 扁平化多级双向链表

原题地址： https://leetcode-cn.com/probl...
repo地址： https://github.com/pigpigever...

遇到问题咱们首先要先搞清楚问题究竟是什么，而后再想办法解决。对于 链表 的问题其实大部分都不算难，写代码以前最好动动笔在纸上画一下，思路便会清晰不少。

题目剖析🧐

题目中描述这样一个结构的链表：

function Node(val,prev,next,child) {
    this.val = val;
    this.prev = prev;
    this.next = next;
    this.child = child;
}

因此这不单单是一个 双向 链表，同时它还可能存在 子链表

咱们再来看看官方提供的输入输出例子👇

输入：

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

输出:

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

扁平化的步骤能够当作这样：

第一步：

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

第二步：

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

显然，咱们须要递归处理 子链表，再把它跟 父链表 拼接在一块儿。

梳理逻辑💡

分析完题目以后咱们来梳理下逻辑👇

• 找到 子链表 而且把它拼接到 父链表 的 下一个节点 上
• 若是 存在 子子链表，那么先处理 子子链表

拼接的逻辑能够更加细化，以下👇

• 保存 子链表 的 第一个节点
• 保存 子链表 的 前一个节点 和这个节点的 下一个节点
• 保存 子链表 的 最后一个节点
• 有了 子链表 的 首尾节点 和 先后节点， 就能够把 子链表 拼接到 父链表 上

示例代码🌰

/**
 * @param {Node} head
 * @return {Node}
 */
var flatten = function(head) {
    dfs(head)
    function dfs (head) {
        let target = head, pre = null, last = null
        while (target) {
            let next = target.next, child = target.child
            if (child) {
                target.next = child
                child.prev = target
                target.next = child
                last = dfs(child)
                last.next = next
                pre = last
                target.child = null
                if (next) {
                    next.prev = last
                }
            } else {
                pre = target
            }
            target = next
        }
        return pre
    }
    return head
};

固然这里有个须要注意的地方，next 、 prev 和 child 指针均可能是空的，写代码的时候须要把这些状况都考虑进去。

写在最后🔚

一直在 LeetCode 上刷题，以前还加入了组织，有兴趣加入一块儿学习的同窗能够在下方留言或者关注个人微信公众号「tony老师的前端补习班」并在后台留言，能够进群跟大佬们一块儿学习。