每日一题

23. 合并K个排序链表

合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6

解1

利用K个指针指向链表的当前位置,每次循环找出最小值,复杂度O(KN)

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        ListNode ans = new ListNode(0);
        ListNode cur = ans;
        int len = lists.length;
        
        while(true){
            ListNode tmp = null;
            int index = -1;
            for(int i=0;i<len;i++){
                if(lists[i]==null)
                continue;
                if(tmp==null||tmp.val>lists[i].val){
                    tmp=lists[i];
                    index=i;
                }
            }

            if(index==-1)
            break;
            cur.next = tmp;
            cur=cur.next;
            lists[index]=lists[index].next;
        }
        return ans.next;
    }
}




解2

利用小顶堆的思想,简化解1 中的复杂度 O(NlogK)

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        ListNode ans = new ListNode(0);
        ListNode cur = ans;
        Queue<ListNode> que = new PriorityQueue<>((o1,o2)->{
            return o1.val-o2.val;
        });
        for(ListNode tr:lists){
            if(tr!=null)
            que.offer(tr);
        }
        while(!que.isEmpty()){
           ListNode tmp= que.poll();
           cur.next=tmp;
           cur=cur.next;
           if(tmp.next!=null){
               que.offer(tmp.next);
           }
        }
        return ans.next;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值