NKOJ 3765 k个最小和
时间限制 : - MS 空间限制 : 65536 KB 该题不设总的时间(空间)限制,每组测试数据具有独立的时空限制
评测说明 : 时限1000ms
问题描述
有k个整数数组,各包含k个元素,从每个数组中选取一个元素加起来,可以得到k^k个和,求这些和中最小的k个值。
输入格式
第一行,一个整数k(k<=500)
接下来k行,每行k个正整数(<=1000000)
输出格式
一行,k个有小到大排列的整数,表示最小的k个和
样例输入 1
3
1 8 5
9 2 5
10 7 6
样例输出 1
9 10 12
样例输入 2
2
1 1
1 2
样例输出 2
2 2
提示
样例1说明:
选出的三组数分别是(1 2 6) (1 2 7) (1 5 6)
来源 改编自uva11997
思路:
先考虑两个数组两两合并得到小k个数的情况,将第三个数组与前面合并得的数组再次合并,便可以解决问题。
两两合并的方法:
a、b排序后,先将a[i]+b[1]的值压入堆中。每次取出一个最小值,作为加入新序列中,再将a[i]+b[j+1]的值压入堆中。
小优化:
只存b的编号,a[i]+b[j+1]的值为 当前值-b[j]+b[j+1];
代码:
#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
const int need=502;
struct fy
{
int num,id;
fy(int a,int b) {num=a,id=b;}
friend bool operator< (fy a,fy b)
{
return a.num>b.num;
}
};
int k;
int a[need]={0,1,2,3,4,5,6,7,8},b[need]={0,1,2,3,4,5,6,7,8};
void merge()
{
priority_queue<fy> q;
if(k==0) puts("dddd");
for(int i=1;i<=k;i++)
q.push(fy(a[i]+b[1],1));
for(int i=1,num,id;i<=k;i++)
{
num=q.top().num,id=q.top().id;
q.pop();
a[i]=num;
if(id<k) q.push(fy(num-b[id]+b[id+1],id+1));
}
}
int main()
{
scanf("%d",&k);
for(int i=1;i<=k;i++) scanf("%d",&a[i]);
sort(&a[1],&a[k+1]);
for(int i=1,j;i<k;i++)
{
for(j=1;j<=k;j++) scanf("%d",&b[j]);
sort(&b[1],&b[k+1]);
merge();
}
for(int i=1;i<=k;i++) printf("%d ",a[i]);
}