23. 合并 K 个升序链表
  1BnnW8rtw7M9 2023年11月02日 31 0


23. 合并 K 个升序链表

优先级队列

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        ListNode dummy = new ListNode(-1), d = dummy;
        PriorityQueue<ListNode> pq = new PriorityQueue<>((a, b) -> Integer.compare(a.val, b.val));
        for(ListNode head : lists){
            if(head != null) pq.offer(head);
        }

        while(!pq.isEmpty()){
            ListNode cur = pq.poll();
            d.next = cur;
            d = d.next;
            if(cur.next != null) pq.offer(cur.next);
        }

        return dummy.next;
    }
}

分治

归并排序思想

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        return merge(lists, 0, lists.length - 1);
    }

    ListNode merge(ListNode[] lists, int l, int r){
        if(l > r) return null;
        if(l == r) return lists[l];

        int mid = (l + r) >> 1;
        return mergeTwoLists(merge(lists, l, mid), merge(lists, mid + 1, r));
    }

    ListNode mergeTwoLists(ListNode list1, ListNode list2){
        ListNode l1 = list1, l2 = list2;
        ListNode dummy = new ListNode(-1), d = dummy;

        while(l1 != null && l2 != null){
            if(l1.val < l2.val){
                d.next = l1;
                l1 = l1.next;
            }else{
                d.next = l2;
                l2 = l2.next;
            }
            d = d.next;
        }
        d.next = l1 == null ? l2 : l1;

        return dummy.next;
    }
}


【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
1BnnW8rtw7M9
作者其他文章 更多