[leetcode/lintcode 題解] 位元組跳動面試題:合併k個排序陣列
Input:
[
[1, 3, 5, 7],
[2, 4, 6],
[0, 8, 9, 10, 11]
]
Output: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
樣例 2:
Input:
[
[1,2,3],
[1,2]
]
Output: [1,1,2,2,3]
演算法一 暴力
題目指出k個數組是有序的,那我們可以借鑑歸併排序的思想
每次遍歷k個有序陣列的陣列第一個元素,找出最小的那個,然後壓入答案ans陣列,記得把最小的那個從它原先的陣列給刪除- 先將每個有序陣列的第一個元素壓入優先佇列中
- 不停的從優先佇列中取出最小元素(也就是堆頂),再將這個最小元素所在的有序陣列的下一個元素壓入佇列中 eg. 最小元素為x,它是第j個數組的第p個元素,那麼我們把第j個數組的第p+1個元素壓入佇列
public class Solution {
/**
* @param arrays: k sorted integer arrays
* @return: a sorted array
*/
static class Node implements Comparator<Node> {
public int value;
public int arrayIdx;
public int idx;
public Node() {
}
//value權值大小,arraysIdx在哪個數組裡,idx在該陣列的哪個位置> >
public Node(int value, int arrayIdx, int idx) {
this.value = value;
this.arrayIdx = arrayIdx;
this.idx = idx;
}
public int compare(Node n1, Node n2) {
if(n1.value < n2.value) {
return 1;
} else {
return 0;
}
}
}
static Comparator<Node> cNode = new Comparator<Node>() {
public int compare(Node o1, Node o2) {
return o1.value - o2.value;
}
};
public int[] mergekSortedArrays(int[][] arrays) {
// 初始化 優先佇列 ,我們優先佇列的一個元素包括三個值 :數字大小,數字在哪個數組裡,數字在陣列的哪個位置
PriorityQueue<Node> q = new PriorityQueue<Node>(arrays.length + 5, cNode);
// 初始化 答案
List<Integer> ans = new ArrayList<>();
for(int i = 0; i < arrays.length; i++) {
// 如果這個陣列為空 則不用壓入
if(arrays[i].length == 0) {
continue;
}
// arrays[i][0] 權值大小 i 在第i個數組 0 在該陣列的0位置
q.add(new Node(arrays[i][0], i, 0));
}
while(!q.isEmpty()) {
// 取出佇列中最小值
Node point = q.poll();
// 權值 ,所在陣列的編號,在該陣列的位置編號
int value = point.value;
int arrayIdx = point.arrayIdx;
int idx = point.idx;
// 更新答案陣列
ans.add(value);
// 它已經是所在陣列的最後一個元素了,這個陣列的所有元素都已經處理完畢
if(idx == arrays[arrayIdx].length - 1) {
continue;
} else {
// 壓入它下一個位置的新元素
Node newPoint = new Node(arrays[arrayIdx][idx + 1], arrayIdx, idx + 1);
q.add(newPoint);
}
}
return ans.stream().mapToInt(Integer::valueOf).toArray();
}
}
更多題解參考:九章官網solution
相關推薦
[leetcode/lintcode 題解] 位元組跳動面試題:合併k個排序陣列
將 k 個有序數組合併為一個大的有序陣列。 線上評測地址:領釦題庫官網 樣例 1:
[leetcode/lintcode 題解] 位元組跳動面試題:換硬幣
給出不同面額的硬幣以及一個總金額. 寫一個方法來計算給出的總金額可以換取的最少的硬幣數量. 如果已有硬幣的任意組合均無法與總金額面額相等, 那麼返回 -1.
[leetcode/lintcode 題解]阿里巴巴面試題:最大子陣列 II
技術標籤:演算法leetcodepythonjava資料結構 描述 給定一個整數陣列,找出兩個 不重疊 子陣列使得它們的和最大。 每個子陣列的數字在陣列中的位置應該是連續的。 返回最大的和。
【LeetCode/LintCode】丨Google面試題:N皇后問題
n皇后問題是將n個皇后放置在n*n的棋盤上,皇后彼此之間不能相互攻擊(任意兩個皇后不能位於同一行,同一列,同一斜線)。
位元組跳動面試題:你的平均薪水是多少?
薪水錶中記錄了員工的編號,所在部門編號,和薪水。 查詢出每個部門除去最高、最低薪水後的平均薪水,並保留整數。(位元組跳動面試題)
leetcode 精選top面試題 - 88. 合併兩個有序陣列
88. 合併兩個有序陣列 給你兩個有序整數陣列nums1 和 nums2,請你將 nums2 合併到nums1中,使 nums1 成為一個有序陣列。
劍指offer程式碼解析——面試題17合併兩個排序的連結串列
本題的詳細解析均在程式碼註釋中: /** * 題目:輸入兩個遞增排序的連結串列,合併這兩個連結串列並使新連結串列中的結點仍然是按照遞增排序的。
【LeetCode/LintCode】 題解丨阿里巴巴面試題:第k大元素
在陣列中找到第 k 大的元素。(你可以交換陣列中的元素的位置) 線上評測地址:點選此處前往
【LeetCode/LintCode】 題解丨Facebook面試題:序列重構
判斷是否序列 org 能唯一地由 seqs重構得出. org是一個由從1到n的正整數排列而成的序列,1≤n≤10^4。 重構表示組合成seqs的一個最短的父序列 (意思是,一個最短的序列使得所有 seqs裡
【LeetCode/LintCode】題解丨 Google面試題:單詞搜尋 II
給出一個由小寫字母組成的矩陣和一個字典。找出所有同時在字典和矩陣中出現的單詞。一個單詞可以從矩陣中的任意位置開始,可以向左/右/上/下四個相鄰方向移動。一個字母在一個單詞中只能被使用一次。且字典中不存在重
牛客網--位元組跳動面試題--雀魂啟動
牛客網--位元組跳動面試題--雀魂啟動 部落格說明 文章所涉及的資料來自網際網路整理和個人總結,意在於個人學習和經驗彙總,如有什麼地方侵權,請聯絡本人刪除,謝謝!
牛客網--位元組跳動面試題--萬萬沒想到之聰明的編輯
牛客網--位元組跳動面試題--萬萬沒想到之聰明的編輯 部落格說明 文章所涉及的資料來自網際網路整理和個人總結,意在於個人學習和經驗彙總,如有什麼地方侵權,請聯絡本人刪除,謝謝!
牛客網--位元組跳動面試題--萬萬沒想到之抓捕孔連順
牛客網--位元組跳動面試題--萬萬沒想到之抓捕孔連順 部落格說明 文章所涉及的資料來自網際網路整理和個人總結,意在於個人學習和經驗彙總,如有什麼地方侵權,請聯絡本人刪除,謝謝!
牛客網--位元組跳動面試題--特徵提取
牛客網--位元組跳動面試題--特徵提取 部落格說明 文章所涉及的資料來自網際網路整理和個人總結,意在於個人學習和經驗彙總,如有什麼地方侵權,請聯絡本人刪除,謝謝!
【位元組跳動面試題】 擼它
牛客刷題偶遇 位元組後端筆試0715 擼它 一共兩道程式設計題。 1、 給定一個合法的計算表示式,該表示式只包括
位元組跳動面試題(封裝一個佇列)
技術標籤:面試題javascript function ArrayQueue(){ var arr = []; //入隊操作 this.push = function(element){
位元組跳動面試題
技術標籤:js手寫題目javascript 駝峰轉換: <!DOCTYPE html> <html lang="en">
位元組跳動面試題,給你一個每一項都是數值混亂順序的陣列,只要裡面正確順序的值輸出。如[5,1,3,6,2,7],只要[1,2,7]
//位元組跳動面試題,給你一個每一項都是數值混亂順序的陣列,只要裡面正確順序的值輸出,如[5,1,3,6,2,7],只要[1,2,7]
《劍指offer》面試題12:合併兩個排序的連結串列
題目描述 輸入兩個單調遞增的連結串列,輸出兩個連結串列合成後的連結串列,當然我們需要合成後的連結串列滿足單調不減規則。
微軟面試題:劍指 Offer 51. 陣列中的逆序對 Hard 出現次數:3
題目描述: 在陣列中的兩個數字,如果前面一個數字大於後面的數字,則這兩個數字組成一個逆序對。