23. Merge k Sorted Lists

最新推荐文章于 2021-12-08 15:59:42 发布

原创最新推荐文章于 2021-12-08 15:59:42 发布 · 393 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

558 篇文章

订阅专栏

本文介绍了一种使用优先队列合并K个已排序链表的方法，并提供了完整的Java实现代码。该方法首先初始化一个优先队列，然后遍历所有输入链表并将它们的第一个节点加入队列。接下来，通过循环从队列中取出最小节点并将其添加到结果链表中，如果该节点后还有节点，则继续将下一个节点放入队列。此过程重复直至队列为空。

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

idea：use priority queue，it is provided by util package

package l23;

import java.util.Comparator;
import java.util.PriorityQueue;

public class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
    	
    	if(lists.length == 0)	return null;
    	
    	PriorityQueue<ListNode> q = new PriorityQueue<ListNode>(lists.length, 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.val < o2.val)	return -1;
				else 						return 0;
			}
    	});
    	
        ListNode root = new ListNode(0);
        ListNode cur = root;
        
        for(ListNode node : lists)
        	if(node != null)
        		q.add(node);
        
        while(!q.isEmpty()) {
        	cur.next = q.poll();
        	cur = cur.next;
        	
        	if(cur.next != null)
        		q.add(cur.next);
        }
        
    	return root.next;
    }
}