LeetCode——430. 扁平化多级双向链表（Java）

Posted 2021-09-24 21丶01丶15

题目描述

题干：
多级双向链表中，除了指向下一个节点和前一个节点指针之外，它还有一个子链表指针，可能指向单独的双向链表。
这些子列表也可能会有一个或多个自己的子项，依此类推，生成多级数据结构，如下面的示例所示。
给你位于列表第一级的头节点，请你扁平化列表，使所有结点出现在单级双链表中。

示例 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]

示例 3：
输入：head = []
输出：[]

---

题解思路

返回扁平化处理的双向链表，我们只需要把每个有 child 的节点的分支链表插入到他的母链表里即可

所以我们需要依次便利每个有 child 的节点，之后将分支链表的尾节点连接到 next

将当前的 next 连接到 child，最后将 child 置空。这样就完成了一次扁平化

---

正确代码

    public Node flatten(Node head) {
        dfs(head);
        return head;
    }

    private Node dfs(Node node) {
        Node cur = node;
        Node last = null;

        while (cur != null) {
            Node next = cur.next;

            // 如果有子节点，先遍历子节点
            if (cur.child != null) {
                Node childLast = dfs(cur.child);

                next = cur.next;
                // 将 node 与 child 相连
                cur.next = cur.child;
                cur.child.prev = cur;

                // 如果 next 不为空，就将 last 与 next 相连
                if (next != null) {
                    childLast.next = next;
                    next.prev = childLast;
                }

                // 将child置空
                cur.child = null;
                last = childLast;
            } else {
                last = cur;
            }
            cur = next;
        }
        return last;
    }

    class Node {
        public int val;
        public Node prev;
        public Node next;
        public Node child;
    }

---

总结

双向链表和普通的链表相差不大，只需要注意插入节点的顺序即可，当然也可以采用递归解决

如果文章存在问题或者有更好的题解，欢迎在评论区斧正和评论，各自努力，你我最高处见