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

[题目链接](剑指 Offer II 028. 展平多级双向链表 - 力扣（LeetCode） (leetcode-cn.com))

思路

• 递归，记住插入规则
  

代码

/*
// Definition for a Node.
class Node {
    public int val;
    public Node prev;
    public Node next;
    public Node child;
};
*/

class Solution {
    public Node flatten(Node head) {
        getTail(head);
        return head;
    }
    private Node getTail(Node head) {//找到该结点的尾结点
        Node node = head;
        Node tail =null;
        while(node != null) {
            Node child = node.child;
            Node next = node.next;
            if(child != null) {
                Node childTail = getTail(child);//找到该孩子节点的尾结点

                childTail.next = next;//孩子节点的尾结点的下一个节点是node节点的下一个节点
                if(next != null) next.prev = childTail;如果next节点不是空节点，就把孩子节点的尾结点的下一个节点的前指针指向孩子节点
                node.next = child;
                child.prev = node;
                node.child = null;//插入之后将node节点的孩子节点变成空节点

                tail = childTail;
            } else {//如果没有孩子节点
                tail = node;//尾结点为当前node节点
            }
            node = next;
        }
        return tail;
    }
}