Leetcode-23: Merge k Sorted Lists

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

将k个有序的列表归并成一个有序列表。

思路:数据结构中学过的是k=2时的特例,不过实际都是每次都将所有列表中头元素最小的那个元素加入新的列表。有两种方法:
1)朴素的比较然后归并方法,对于每个新加入的元素都要做比较k次,那么复杂度将是 O(nk) ,这里的比较是重复的,即加入第一个元素的时候, a0 , a1 , …, ak1 ,做了 k 次比较,选出最小的元素,假设为ai,第二次加入的时候,所有的头节点元素变成了 a0 , a1 , …, bi , …, ak1 ,除了 bi 之外,其他元素都是已经做过比较的了,无需再做多余的比较,所以可以利用优先队列(最小堆)来维护这k个元素之间的顺序。建堆复杂度为 O(k) ,每次加入新元素的复杂度为 O(lgk) ,所以总的复杂度为 O(k+(nk)lgk+klgk)=O(nlgk+k)

2)由于 k>2 时会带来重复比较,那么可以将 k 转化为k=2的问题,即使用分而治之的方法,将列表两两归并,最终归并成一条列表。 k=2 时,归并的复杂度为 O(n) ,两两归并的次数共 lgk 次,所以复杂度为 O(nlgk)

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) return null;
        if (lists.length == 1) return lists[0];
        Queue<ListNode> pq = new PriorityQueue<>(lists.length, new Comparator<ListNode>(){
            @Override
            public int compare(ListNode l1, ListNode l2) {
                if (l1.val < l2.val)
                    return -1;
                else if (l1.val > l2.val)
                    return 1;
                else
                    return 0;
            }
        });
        ListNode sential = new ListNode(0);
        for (ListNode list : lists){
            if (list != null)
                pq.add(list);
        }

        ListNode tail = sential;
        while (!pq.isEmpty()) {
            tail.next = pq.poll();
            tail = tail.next;
            if (tail.next != null)
                pq.add(tail.next);
        }
        return sential.next;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值