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(Python)

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

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...
  • jmspan
  • jmspan
  • 2016年07月14日 00:51
  • 424

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...
  • zshouyi
  • zshouyi
  • 2017年05月22日 15:57
  • 82

leetcode Find K Pairs with Smallest Sums

leetcode 373 find K pairs with Smallest Sums

UVa - 11997 K Smallest Sums(优先队列多路归并)

题意是k个数组,每个数组k个元素,每个数组取一个元素,共k个数之和成为一个参考值,求出k^k个参考值中最小的k个    ... 方法就是先 对每个数组排序,    然后由于k个之和最小和 2...
  • viphong
  • viphong
  • 2015年09月04日 15:51
  • 259

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),还能骗...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:leetcode_373 Find K Pairs with Smallest Sums
举报原因:
原因补充:

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