前言
此题是自己在整理算法题目时候发现字节的面试官经常会问道这个题目,这里就做一个记录。
首先来看一下对应的leetcode:
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
思考
合并链表
首先这个题目是对于合并两个排序链表或者是说排序的数组的一个变形处理,这里就不讲述具体的合并两个的行为。关于具体的代码可以参见解体答案,这里利用到了优先队列:
来看具体的实现:
首先我们利用到优先队列:
Queue<ListNode> pq = new PriorityQueue<>(new Comparator<ListNode>() {
@Override
public int compare(ListNode o1, ListNode o2) {
return o1.val-o2.val;
}
});
// 表示在使用到poll()时候都会抛出当前链表头节点的最小值。
public ListNode mergeKLists(ListNode[] lists) {
Queue<ListNode> pq = new PriorityQueue<>(new Comparator<ListNode>() {
@Override