1. 題目

給定兩個(gè)以升序排列的整形數(shù)組 nums1 和 nums2, 以及一個(gè)整數(shù) k。

定義一對(duì)值 (u,v)，其中第一個(gè)元素來自 nums1，第二個(gè)元素來自 nums2。

找到和最小的 k 對(duì)數(shù)字 (u1,v1), (u2,v2) … (uk,vk)。

示例 1: 輸入: nums1 = [1,7,11], nums2 = [2,4,6], k = 3 輸出: [1,2],[1,4],[1,6] 解釋: 返回序列中的前 3 對(duì)數(shù)：[1,2],[1,4],[1,6],[7,2],[7,4],[11,2],[7,6],[11,4],[11,6]示例 2: 輸入: nums1 = [1,1,2], nums2 = [1,2,3], k = 2 輸出: [1,1],[1,1] 解釋: 返回序列中的前 2 對(duì)數(shù)：[1,1],[1,1],[1,2],[2,1],[1,2],[2,2],[1,3],[1,3],[2,3]示例 3: 輸入: nums1 = [1,2], nums2 = [3], k = 3 輸出: [1,3],[2,3] 解釋: 也可能序列中所有的數(shù)對(duì)都被返回:[1,3],[2,3]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/find-k-pairs-with-smallest-sums
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。

2. 解題

• 類似題目： LeetCode 5403. 有序矩陣中的第 k 個(gè)最小數(shù)組和 hard

上面鏈接5403題目是n行，本題是2行，本質(zhì)并無區(qū)別

• 每行一個(gè)指針，初始都位于最前面
• 優(yōu)先隊(duì)列存儲(chǔ)《和，指針1，指針2》，自定義和小的優(yōu)先
• 每次將堆頂?shù)摹吨羔槍?duì)》取出，依次對(duì)每一行指針+1，數(shù)值變大一點(diǎn)
• push進(jìn)入優(yōu)先隊(duì)列（堆），重復(fù)以上過程k次

struct cmp {bool operator()(vector<int>& a, vector<int>& b){return a[0] > b[0];//小頂堆，和小的在上面} }; class Solution { public:vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {int i1 = 0, i2 = 0, n1 = nums1.size(), n2 = nums2.size();if(k > n1*n2) k = n1*n2;if(n1==0 || n2==0) return {};priority_queue<vector<int>, vector<vector<int>>, cmp> q;q.push({nums1[0]+nums2[0],0,0});//sum,i1,i2vector<vector<int>> ans;vector<vector<bool>> visited(n1,vector<bool>(n2,false));visited[0][0] = true;while(k--){i1 = q.top()[1];i2 = q.top()[2];ans.push_back({nums1[i1],nums2[i2]});q.pop();if(i1+1 < n1 && !visited[i1+1][i2]){q.push({nums1[i1+1]+nums2[i2],i1+1,i2});visited[i1+1][i2] = true;}if(i2+1 < n2 && !visited[i1][i2+1]){q.push({nums1[i1]+nums2[i2+1],i1,i2+1});visited[i1][i2+1] = true;}}return ans;} };

20 ms 9.9 MB (本題數(shù)據(jù)量變大，上面解法失效)

• 初始把 數(shù)組1 中的前 min(len(nums1), k) 個(gè)元素 和 nums2[0] 配對(duì)加入優(yōu)先隊(duì)列
• 在優(yōu)先隊(duì)列內(nèi)只需對(duì) 指針2 進(jìn)行 +1 枚舉

struct cmp {bool operator()(vector<long long>& a, vector<long long>& b){return a[0] > b[0];} }; class Solution { public:vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {int i1 = 0, i2 = 0, n1 = nums1.size(), n2 = nums2.size();if(k > 1LL*n1*n2) k = n1*n2;if(n1==0 || n2==0) return {};priority_queue<vector<long long>, vector<vector<long long>>, cmp> q;for (int i = 0; i < min(k, n1); i++) q.push({nums1[i]+nums2[0],i,0});//sum,i1,i2vector<vector<int>> ans;while(k--){i1 = q.top()[1];i2 = q.top()[2];ans.push_back({nums1[i1],nums2[i2]});q.pop();if(i2+1 < n2){q.push({nums1[i1]+nums2[i2+1],i1,i2+1});}}return ans;} };

執(zhí)行用時(shí)：100 ms, 在所有 C++ 提交中擊敗了52.91%的用戶
2022.1.14

總結(jié)

以上是生活随笔為你收集整理的LeetCode 373. 查找和最小的K对数字（自定义优先队列BFS）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。