题意:给出k个长度为k的整数数组,每个数组里面取一个元素加起来,共有k^k个和,输出这些和的前k个(升序排序后)。
经典的多路合并问题,用一个优先队列维护k个值,每次pop出最小值之后将其下一个值加入队列。但是这题有k个数组,我们可以每次合并两个数组,求出前k个(后面的值我们肯定用不着了),接着再和下一个数组合并。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int maxn = 1000;
int arr1[maxn], arr2[maxn], k;
struct pp
{
int s, b;
pp(int aa, int bb) : s(aa), b(bb) { }
bool operator < (const pp & rhs) const { return s > rhs.s; }
};
int main()
{
freopen("in.txt", "r", stdin);
while(~scanf("%d", &k))
{
for(int i = 0; i < k; ++i) scanf("%d", arr1 + i);
sort(arr1, arr1 + k);
for(int i = 1; i < k; ++i)
{
for(int j = 0; j < k; ++j