题解 | #合并k个已排序的链表#

合并k个已排序的链表

https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6

alt alt step 1:从链表数组的首和尾开始,每次划分从中间开始划分,划分成两半,得到左边n/2个链表和右边n/2个链表。 step 2:继续不断递归划分,直到每部分链表数为1. step 3:将划分好的相邻两部分链表,按照两个有序链表合并的方式合并,合并好的两部分继续往上合并,直到最终合并成一个链表。

分治法和双指针

            return null;
        }
        if (left == right) {
            return lists.get(left);
        }
        int mid = (left + right) / 2;
        return Merge(divideMerge(lists, left, mid), divideMerge(lists, mid + 1, right));

对于这k个链表,就相当于上述合并阶段的k个子问题,需要划分为链表数量更少的子问题,直到每一组合并时是两两合并,然后继续往上合并,这个过程基于递归:

终止条件: 划分的时候直到左右区间相等或左边大于右边。 返回值: 每级返回已经合并好的子问题链表。 本级任务: 对半划分,将划分后的子问题合并成新的链表。

        if (pHead1 == null && pHead2 == null)return null;
        if (pHead1 != null && pHead2 == null)return pHead1;
        if (pHead1 == null && pHead2 != null)return pHead2;
        ListNode head = new ListNode(0);
        ListNode cur = head;
        while (pHead1 != null && pHead2 != null) {
            if (pHead1.val < pHead2.val) {
                cur.next = pHead1;
                cur = cur.next;
                pHead1 = pHead1.next;
            } else {
                cur.next = pHead2;
                cur = cur.next;
                pHead2 = pHead2.next;
            }
        }
        if (pHead1 != null)cur.next = pHead1;
        if (pHead2 != null)cur.next = pHead2;
        return head.next;
    }
    ListNode divideMerge(ArrayList<ListNode> lists, int left, int right) {
        if (left > right) {
            return null;
        }
        if (left == right) {
            return lists.get(left);
        }
        int mid = (left + right) / 2;
        return Merge(divideMerge(lists, left, mid), divideMerge(lists, mid + 1, right));
    }
    public ListNode mergeKLists (ArrayList<ListNode> lists) {
        // write code here
        return divideMerge(lists, 0, lists.size() - 1);

    }

参考题解:https://blog.nowcoder.net/n/0a37bd3344eb499e88697dc648f00b62

全部评论

相关推荐

野猪不是猪🐗:这种直接口头上答应,骗面试,面完了直接拉黑,相当于给自己攒面经了(
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务