提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
题目描述
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 1:
输入:lists = [[1,4,5],[1,3,4],[2,6]]
输出:[1,1,2,3,4,4,5,6]
解释:链表数组如下:
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6
示例 2:
输入:lists = []
输出:[]
示例 3:
输入:lists = [[]]
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-k-sorted-lists
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题过程
解题思路
整体思路:首先取出每一个链表的头结点,找到这些头结点中的最小值,同时记录最小值对应的是哪一条链表,然后用这个最小值创建新链表的结点,将最小值对应的链表向后移动。找头结点中的最小值时要注意两个地方:
1、链表是否为空;
2、要记录空链表的个数。
代码中死循环退出的条件是空链表个数等于链表数组的长度。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
ListNode cur = new ListNode(0);
ListNode res = cur;
int listSize = lists.length;
while(true){
int min = Integer.MAX_VALUE;
int whichList = 0;
int breakFlag = 0;
for(int i = 0; i < listSize; i++){
if(lists[i] == null){
breakFlag++;
}
if(lists[i] == null){
continue;
}
if(lists[i].val < min){
min = lists[i].val;
whichList = i;
}
}
if(breakFlag == lists.length){
break;
}
lists[whichList] = lists[whichList].next;
cur.next = new ListNode(min);
cur = cur.next;
}
return res.next;
}
}