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 解题报告

题目链接: https://leetcode.com/problems/find-k-pairs-with-smallest-sums/ You are given two integer arr...
  • qq508618087
  • qq508618087
  • 2016年07月09日 01:32
  • 4216

[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...
  • TstsUgeg
  • TstsUgeg
  • 2016年07月08日 17:11
  • 1109

【Leetcode】Find K Pairs with Smallest Sums

题目链接:https://leetcode.com/problems/find-k-pairs-with-smallest-sums/ 题目: You are given two integer ...
  • yeqiuzs
  • yeqiuzs
  • 2016年07月07日 15:13
  • 3947

UVA K Smallest Sums(多路归并,优先队列)

Description You're given k arrays, each array has k integers. There are kk ways to pick exactly o...
  • Misdom_Tian_Ya
  • Misdom_Tian_Ya
  • 2015年04月23日 10:33
  • 452

Find K Pairs with Smallest Sums

You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k. Define...
  • gettogetto
  • gettogetto
  • 2016年07月07日 22:57
  • 1040

UVA 11997--K Smallest Sums+优先队列用于多路归并

题目链接:点击进入 先考虑两个数组A,B的情况,这样总共有n^2种情况;将A,B数组排序后,我们可以将所有情况组织成n张表: 表1: A[1]+B[1]...
  • acm_lkl
  • acm_lkl
  • 2015年06月13日 13:12
  • 789

UVA11997:K Smallest Sums(多路归并求最小k个和)

You’re given k arrays, each array has k integers. There are k k ways to pick exactly one element in...
  • libin56842
  • libin56842
  • 2015年06月10日 20:26
  • 1038

[LeetCode算法笔记]Find K Pairs with Smallest Sums与优先队列

LeetCode第373题 Find K Pairs with Smallest Sums是一道中等难度的题。给定两组升序排列的整形(int)数组,从中分别挑选k对和最小的数对(pair),意思就是从...
  • tostq
  • tostq
  • 2016年07月13日 14:12
  • 1311

[LeetCode]373. Find K Pairs with Smallest Sums

https://leetcode.com/problems/find-k-pairs-with-smallest-sums/?tab=Description 给两个升序数组,找到和最小的k个pa...
  • gqk289
  • gqk289
  • 2017年03月08日 20:52
  • 68

Leetcode: Find K Pairs with Smallest Sums

You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k.Define a...
  • loushuai
  • loushuai
  • 2016年09月16日 08:07
  • 112
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:leetcode_373 Find K Pairs with Smallest Sums
举报原因:
原因补充:

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