题目
给定两个以 升序排列 的整数数组 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]
解题
方法一:优先队列
换成array
运行效率比vector
更高,因为array
是固定长度的(同理也可以换成pair
)
class Solution { public: vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) { auto cmp=[&](array<int,2>& a,array<int,2>& b){ return nums1[a[0]]+nums2[a[1]]>nums1[b[0]]+nums2[b[1]]; }; priority_queue<array<int,2>,vector<array<int,2>>,decltype(cmp)> q(cmp); for(int i=0;i<nums1.size();i++){ q.push({i,0}); } vector<vector<int>> res; while(k--&&!q.empty()){ array<int,2> pos=q.top(); q.pop(); res.push_back({nums1[pos[0]],nums2[pos[1]]}); if(++pos[1]<nums2.size()){ q.push({pos[0],pos[1]}); } } return res; } };