LCR 028

题目LCR 028


解法一:深度优先搜索

遍历链表,若某节点存在子链,则遍历子链,遍历完后,返回子链最后一个节点 last,将 last 与当前链表的 next 相连

    public Node flatten(Node head) {
        dfs(head);
        return head;
    }
    public Node dfs(Node head) {
        Node cur = head, last = null, next = null;
        while (cur != null) {
            last = cur;
            next = cur.next;
            //连接孩子节点,更新last,并把last和next相连
            if (cur.child != null) {
                //连接child
                cur.next = cur.child;
                cur.child.prev = cur;
                //获取子链尾节点
                last = dfs(cur.child);
                //将子节点置空
                cur.child = null;
                //将子链尾节点和下一个节点相连
                last.next = next;
                if (next != null) next.prev = last;
            }
            //此处不能用cur=cur.next,因为此时cur的next已经被修改为孩子节点
            cur = next;
        }
        return last;
    }

注意

  • next为空时,只需要将lastnext指针指向next,不需要将nextprev指针指向last

  • 扁平化子链后,需要将child节点置空,否则测试不通过


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值