您的位置:首页 > 大数据 > 人工智能

leetcode:heap:Find K Pairs with Smallest Sums(373)

2016-09-01 15:29 393 查看
https://leetcode.com/problems/find-k-pairs-with-smallest-sums/

http://blog.csdn.net/qq508618087/article/details/51864835

class Solution {
public:
vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
int len1 = nums1.size(), len2 = nums2.size(), cnt = min(k, len1*len2);
vector<int> index(len1, 0);
vector<pair<int, int>> ans;
while(cnt-- > 0)
{
int temMin = INT_MAX, m = 0;
for(int i =0; i < len1; i++)
if(index[i] < len2 && nums1[i]+nums2[index[i]] < temMin)
temMin= nums1[i]+nums2[index[i]], m = i;
ans.push_back(make_pair(nums1[m], nums2[index[m]++]));
}
return ans;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: