牛客刷题——合并K个有序列表

题目:

 k 个已排序的链表并将其作为一个已排序的链表返回。分析并描述其复杂度。 

太笨了太笨了,看到有序数列合并还想不到归并排序。

解法一是直接进行归并排序,使用循环进行k-1次归并,

解法二是使用优先队列,由于之前没有用过这种数据结构,因此这里尝试使用优先队列进行解决。

首先使用优先队列的话要重新调整compare方法,使其能够适应ListNode的值的比较。

使用优先队列不需要将所有链表的节点加入,只需要将链表的头节点加入到队列中,队列会根据节点值对队列元素进行排序,然后对队列进行遍历,使用带头节点的res链表来存储最终结果,使用tempRes来作为res的尾指针。在遍历的过程中,取队首元素,添加到res中,队尾指针后移,然后判断当前拿到的这个队首元素还有没有后继节点,有的话添加到队列中,继续执行,直到队列为空。

自己还是缺少利用已知的数结构去解决问题的能力啊。

完整代码如下:

 public ListNode mergeKLists(ArrayList<ListNode> lists) {
		 if(lists==null||lists.size()==0) {
			 return null;
		 }
			PriorityQueue<ListNode> queue = new PriorityQueue<>(lists.size(),new Comparator<ListNode>() {

				@Override
				public int compare(ListNode o1, ListNode o2) {
					// TODO Auto-generated method stub
					if(o1.val>o2.val) {
						return 1;
					}else if(o1==o2) {
						return 0;
					}else {
						return -1;
					}
				}
				
			});
			for(ListNode temp:lists) {
				if(temp!=null) {
					queue.add(temp);
				}
			}
			ListNode res = new ListNode(0);
			ListNode tempRes = res;
			while (!queue.isEmpty()) {
				tempRes.next =  queue.poll();;
				tempRes = tempRes.next;
				if(tempRes.next!=null) {
					queue.add(tempRes.next);
				}
			}
			return res.next;

	 }

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值