解題思路：

使用最大堆來存儲k個和最小的數組組合，然后每進來一個數組組合，只需要和最大堆的堆頂進行比較即可。

用C++的priority_queue實現最大堆時，需要用一下pair來組合來自nums1和nums2數組的數，然后自己寫一個比較結構體cmp，比較數組和的大小。

在循環中的一個trick在于，限制為訪問nums1和nums2的前k個數即可。所以那一部分的時間復雜度為O(k*k)。

?

class Solution { public:vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {if(k == 0) return {{}};priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pq; //最大堆for(int i = 0; i < nums1.size() && i < k; i++){for(int j = 0; j < nums2.size() && j < k; j++){pq.push(make_pair(nums1[i], nums2[j]));if(pq.size() > k) pq.pop();}}//將結果放到res中vector<vector<int> > res;while(!pq.empty()){vector<int> tmp;tmp.push_back(pq.top().first);tmp.push_back(pq.top().second);pq.pop();res.push_back(tmp);}return res;}//自定義比較結構體,比較和的大小struct cmp{bool operator()(pair<int, int> a, pair<int, int> b){if(a.first + a.second < b.first + b.second) return true;return false;}}; };

?

?

?

?

《新程序員》：云原生和全面數字化實踐50位技術專家共同創作，文字、視頻、音頻交互閱讀

總結

以上是生活随笔為你收集整理的Leetcode 373. 查找和最小的K对数字 解题思路及C++实现的全部內容，希望文章能夠幫你解決所遇到的問題。

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