题目描述
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例:
输入: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
题目分析
题干已经给出了K个升序链表,可以利用一个answer链表来维护已经合成的链表,利用for循环合成当前answer与lists中的链表,最后输出answer
题目的关键在于编写一个方法合成两个升序链表
java解法
**
* 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 answer = null;
//循环合成
for(int i = 0; i < lists.length; i++){
answer = mergeTwoLists(answer, lists[i]);
}
return answer;
}
//合并两个升序链表
public ListNode mergeTwoLists(ListNode a, ListNode b){
if(a == null || b == null){
return a != null ? a : b;
}
ListNode pre = new ListNode(0);
ListNode node = pre, aPre = a, bPre = b;
while(aPre != null && bPre != null){
if(aPre.val < bPre.val){
node.next = aPre;
aPre = aPre.next;
}else{
node.next = bPre;
bPre = bPre.next;
}
node = node.next;
}
node.next = (aPre != null ? aPre : bPre);
return pre.next;
}
}