LeetCode 023 Merge K Sorted Lists

题:

https://leetcode.com/problems/merge-k-sorted-lists/

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

解法一:将K个链表做K-1次归并,每次归并是对两个链表的归并,最终得到一个排序的链表。即:

1,2合并,遍历2n个节点;(1,2)与3合并,遍历3n个节点。。。(1...k-1)与k合并,遍历kn个节点。所以总共遍历n*(2+3+...+k)=n*(k^2+k-2)/2,那么时间复杂度为O(n*k^2)。

解法二:对解法一改进一下,改用分治法,所以时间复杂度变为O(nklogk)。

解法三:将K个链表的首元素都取出来,选择出最小的那个作为新链表的head。然后将该元素的next取出来,与其他链表的元素比较再选一个小的,放到新链表中。选择出最小元素的时间复杂度为O(k), 总共要选nk次,所以时间复杂度为O(n*k^2)。

解法四:对解法三改进一下,用最小堆来实现选择最小元素的要求,则时间复杂度降为O(logk),则总的时间复杂度为O(nklogk),与解法二的时间复杂度一样。在Java中,可以使用基于最小堆算法的PriorityQueue来作为最小堆的集合类。代码如下:

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public class Solution {
    public ListNode mergeKLists(List<ListNode> lists) {
        if (lists == null || lists.isEmpty()) {
            return null;
        }
        PriorityQueue<ListNode> minHeap = new PriorityQueue<ListNode>(new Comparator<ListNode>(){
            public int compare(ListNode l1, ListNode l2) {
                return Integer.compare(l1.val, l2.val);
            }
        });
        for (ListNode node : lists) {
            if (node != null) {
                minHeap.add(node);
            }
        }
        ListNode helper = new ListNode(0);
        ListNode next = helper;
        while (!minHeap.isEmpty()) {
            ListNode min = minHeap.poll(); // must not be null
            next.next = min;
            min = min.next;
            if (min != null) {
                minHeap.add(min);
            }
            next = next.next;
        }
        return helper.next;
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值