题解 | #合并k个已排序的链表#
合并k个已排序的链表
https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* public ListNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param lists ListNode类ArrayList
* @return ListNode类
*/
public ListNode mergeKLists (ArrayList<ListNode> lists) {
// write code here
if (lists.size() == 0) {
return null;
}
return mergeSort(lists, 0, lists.size() - 1);
}
public ListNode mergeSort(ArrayList<ListNode> lists, int l, int r) {
if (l == r) {
return lists.get(l);
}
int mid = (l + r) >> 1;
return sort(mergeSort(lists, l, mid), mergeSort(lists, mid + 1, r));
}
public ListNode sort(ListNode p1, ListNode p2) {
ListNode h = new ListNode(0);
ListNode cur = h;
while (p1 != null && p2 != null) {
if (p1.val < p2.val) {
cur.next = p1;
p1 = p1.next;
} else {
cur.next = p2;
p2 = p2.next;
}
cur = cur.next;
}
if (p1 != null) {
cur.next = p1;
}
if (p2 != null) {
cur.next = p2;
}
return h.next;
}
}
使用归并排序的思想
查看18道真题和解析