leetcode_373 Find K Pairs with Smallest Sums

原创 2016年08月29日 19:30:52
  • 题目分析

    给定两个递增的数组,从每个数组中任意取出一个数来组成不同的数字对,求前k个最小的数字对。

  • 解题思路

    方法1 利用暴力求解法;
    方法2 借助STL中的multimap实现;
    方法3 借助STL中的priority_queue实现。

  • 实现程序

static bool cmp(pair<int, int> a, pair<int, int> b)
{
    if (a.first + a.second < b.first + b.second)
        return true;
    return false;
}
// 方法1 暴力求解 
vector< pair<int, int> > kSmallestPairs1(vector<int>& nums1, vector<int>& nums2, int k)
{
    vector< pair<int, int> > result;
    int size1 = (int)nums1.size();
    int size2 = (int)nums2.size();
    if (size1 <= 0 || size2 <= 0)
        return result;
    for (int i = 0; i < min(size1, k); i++)
    {
        for (int j = 0; j < min(size2, k); j++)
        {
            result.push_back(make_pair(nums1[i], nums2[j]));
        }
    } 
    // 对生成的结果进行排序
    sort(result.begin(), result.end(), cmp);
    // 删除重复的元素 
    if (result.size() > k)
        result.erase(result.begin() + k, result.end());
    return result;
} 
// 方法2 借助multimap实现
vector< pair<int, int> > kSmallestPairs2(vector<int>& nums1, vector<int>& nums2, int k)
{
    vector< pair<int, int> > result;
    int size1 = nums1.size();
    int size2 = nums2.size();
    if (size1 <= 0 || size2 <= 0 || k <= 0)
        return result;
    multimap< int, pair<int, int> > mul;
    for (int i = 0; i < min(size1, k); i++)
    {
        for (int j = 0; j < min(size2, k); j++)
        {
            int temp = nums1[i] + nums2[j];
            mul.insert(make_pair(temp, make_pair(nums1[i], nums2[j])));
        }
    }
    // 取出前k个元素
    multimap< int, pair<int, int> >::iterator it;
    for (it = mul.begin(); it != mul.end(); it++)
    {
        result.push_back(it->second);
        k--;
        if (k <= 0)
            return result;
    }
    return result;
}
// 方法3:借助priority_queue实现
struct cmp1
{
    bool operator()(pair<int, int> a, pair<int, int> b)
    {
        return a.first + a.second < b.first + b.second;
    }
};
vector< pair<int, int> > kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k)
{
    vector< pair<int, int> > result;
    int size1 = nums1.size();
    int size2 = nums2.size();
    if (size1 <= 0 || size2 <= 0 || k <= 0)
        return result;
    priority_queue< pair<int, int>, vector< pair<int, int> >, cmp1> que;
    for (int i = 0; i < min(size1, k); i++)
    {
        for (int j = 0; j < min(size2, k); j++)
        {
            if (que.size() < k)
            {
                que.push(make_pair(nums1[i], nums2[j]));
            }
            else
            {
                if (nums1[i] + nums2[j] < que.top().first + que.top().second)
                {
                    que.push(make_pair(nums1[i], nums2[j]));
                    que.pop();
                }
            }
        }

    }
    while (!que.empty())
    {
        result.push_back(que.top());
        que.pop();
    }
    return result;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。 举报

相关文章推荐

[leetcode] 373. Find K Pairs with Smallest Sums

You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k.  Def...

Leetcode 373. Find K Pairs with Smallest Sums(Python)

睡不着咋办!题目大意给定两个数组,定义pair是一个二元组,每个元素分别取自两个数组。求pair和前k小的pair思路用堆,我们维持一个k最大堆即可代码class mHeap(object): ...

我是如何成为一名python大咖的?

人生苦短,都说必须python,那么我分享下我是如何从小白成为Python资深开发者的吧。2014年我大学刚毕业..

Leetcode 373. Find K Pairs with Smallest Sums

Leetcode 373. Find K Pairs with Smallest Sums

[leetcode] 373. Find K Pairs with Smallest Sums

You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k. Defi...

LeetCode 373 Find K Pairs with Smallest Sums

You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k. Define...

LeetCode 373. Find K Pairs with Smallest Sums

原题网址:https://leetcode.com/problems/find-k-pairs-with-smallest-sums/ You are given two integer array...

373. Find K Pairs with Smallest Sums

You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k.  Def...

leetcode Find K Pairs with Smallest Sums

leetcode 373 find K pairs with Smallest Sums

K Smallest Sums(Uva 11997) 多路归并+优先队列

来自《算法竞赛入门经典训练指南》 1.题目原文 https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=s...

UVa11997 K Smallest Sums 归并 + 优先队列

朴素算法O(n ^ n)啊,这都能做,想了想,最后只要输出其中和的前k小的就行了,有很多计算是冗余的啊,想了想,每行排个序,然后只算前两列,还有2 ^ k个,取k个绰绰有余了,O(2 ^ n),还能骗...
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)