給你一個鏈表數組,每個鏈表都已經按升序排列。
請你將所有鏈表合併到一個升序鏈表中,返回合併後的鏈表。
示例 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 = [[]]
輸出:[]
提示:
k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i]
按 升序 排列lists[i].length
的總和不超過10^4
思路
對於多個數組要合併成一條鏈表,首先就是把他們排好序,可以藉助優先級隊列,其次就是把他們取出來,由於放入優先級隊列的只是每個數組的頭節點,所以排序只是針對這三個數,每次取出一個節點放入新節點後,如果這個節點後面還有節點也需要放入隊列中進行排序,依次如這般操作即可組成新鏈表
//leetcode submit region begin(Prohibit modification and deletion)
/**
* 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) {
if(lists==null || lists.length==0){
return null;
}
ListNode dummy = new ListNode(-1);
ListNode p = dummy;
//利用優先級隊列,大數都沉下去
PriorityQueue<ListNode> pq = new PriorityQueue<>(lists.length, (a, b) -> (a.val - b.val));
//PriorityQueue<ListNode> queue = new PriorityQueue<>(
// lists.length, (a, b)->(a.val - b.val));
//放數 放都是每個數組的頭節點
for (ListNode head : lists) {
if(head!=null){
pq.add(head);
//System.out.println(("pq " + head.val));
}
}
//取數構成鏈表,注意每次取的都是每個鏈表的頭節點,取完記得把當前鏈表的下一個節點放進去
while (pq != null && pq.size() != 0) {
ListNode tempNode = pq.poll();
p.next = tempNode;
p = p.next;
if(tempNode!=null&&tempNode.next!=null){
pq.add(tempNode.next);
}
//System.out.println("p: " + p.val);
}
return dummy.next;
}
}
//leetcode submit region end(Prohibit modification and deletion)