思路:
1. 利用数组mark记录nums1中每个元素[i]当前与nums2中关联的位置[ji],
1. 利用数组mark记录nums1中每个元素[i]当前与nums2中关联的位置[ji],
初始关联状态为[0, 0], [1, 0], … , [i, 0], …, [nums1.size()-1,0];
2. 利用小顶堆比较这nums1.size()个状态对应的值( nums1[i]+nums2[ji] ),找出当前的最小值。
2. 利用小顶堆比较这nums1.size()个状态对应的值( nums1[i]+nums2[ji] ),找出当前的最小值。
更新mark[min]的值,并将(nums1[min]+nums2[mark[min]])值加入堆中。
class Solution {
public:
vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
if (nums1.size() == 0 || nums2.size() == 0 )
return retVec;
for (vector<int>::size_type i = 0; i < nums1.size(); ++i){
mark.push_back(0);
que.push(new Node(nums1[i]+nums2[0],i));
}
for (int i = 0; i < k; ++i){
if (que.empty() == false){
int smallestSum = que.top()->sum;
unsigned int pos = que.top()->pos;
que.pop();
int value2=nums2[mark[pos]];
retVec.push_back(make_pair(smallestSum-value2,value2));
mark[pos]++;// 将nums1[pos]关联的nums2中元素位置更新(向后移动一格)
if(mark[pos] >= nums2.size())
continue;
else{
int sum=nums1[pos]+nums2[mark[pos]];
que.push(new Node(sum,pos));
}
}
}
return retVec;
}
private:
class Node{
public:
int sum;// 位于不同数组的两个元素之和
unsigned int pos;// 其中一个元素在nums1中的索引
Node(int s,unsigned int p):sum(s),pos(p){}
};
struct cmp{//用于构建小顶堆
bool operator()(Node* a,Node* b){
return a->sum > b->sum;// 是'>',不是'<'
}
};
vector<int> mark;
priority_queue<Node*,vector<Node*>,cmp> que;
vector<pair<int,int>> retVec;
};