1. 題目

給定兩個以升序排列的整形數組 nums1 和 nums2, 以及一個整數 k。

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

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

示例 1: 輸入: nums1 = [1,7,11], nums2 = [2,4,6], k = 3 輸出: [1,2],[1,4],[1,6] 解釋: 返回序列中的前 3 對數：[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 對數：[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] 解釋: 也可能序列中所有的數對都被返回:[1,3],[2,3]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/find-k-pairs-with-smallest-sums
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

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

上面鏈接5403題目是n行，本題是2行，本質并無區別

• 每行一個指針，初始都位于最前面
• 優先隊列存儲《和，指針1，指針2》，自定義和小的優先
• 每次將堆頂的《指針對》取出，依次對每一行指針+1，數值變大一點
• push進入優先隊列（堆），重復以上過程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 (本題數據量變大，上面解法失效)

• 初始把 數組1 中的前 min(len(nums1), k) 個元素 和 nums2[0] 配對加入優先隊列
• 在優先隊列內只需對 指針2 進行 +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;} };

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。