LeetCode刷题笔录Merge k Sorted Lists

最新推荐文章于 2025-08-22 19:21:44 发布

likecool21

最新推荐文章于 2025-08-22 19:21:44 发布

阅读量664

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode merge

本文链接：https://blog.youkuaiyun.com/likecool21/article/details/38723761

LeetCode 专栏收录该内容

114 篇文章

订阅专栏

本文详细介绍了如何使用优先队列合并多个已排序链表为一个有序链表的过程，时间复杂度为O(nklogk)，其中n为元素总数，k为链表数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

用一个priority queue，先把每个list的第一个node放进去。每次拿一个node出来，并把node.next放进去。事实上排序是由heap来做的，感觉像是作弊的样子。时间复杂度O(nklogk)。n是元素总数。

/**
 * 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) {
    	Comparator<ListNode> comparator = new Comparator<ListNode>() {
			@Override
			public int compare(ListNode o1, ListNode o2) {
				return o1.val - o2.val;
			}
		};
        PriorityQueue<ListNode> queue = new PriorityQueue<ListNode>(50, comparator);
        for(ListNode node : lists){
        	if(node != null)
        		queue.offer(node);
        }
        
        ListNode head = null;
        ListNode pre = null;
        while(!queue.isEmpty()){
        	ListNode node = queue.poll();
        	if(head == null){
        		head = node;
        		pre = node;
        	}
        	else{
        		pre.next = node;
        	}
        	
        	pre = node;
        	if(node.next != null){
        		queue.offer(node.next);
        	}
        }
        
        return head;
    }
}