题意:有n行含m个元素序列,从每行取一个数得到他们的和,一共可以得到m^n个和。输出前n个最小的和。
思路:可以用优先队列递归解决,当只取前两行的数,得到两个数的和的前n小的序列。这个序列就相当于把第一行和第二行合并,再解决n-1行的子问题。
用优先队列解决的时候也有点小技巧,类似尺取法。
//236 KB 563 ms C++ 1480 B
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
struct node
{
int val;
int a,b;
bool operator <(const node &a) const
{
return val>a.val;
}
};
priority_queue<node>que;
const int N = 2000+20;
int a[N],b[N];
int tmp[N];
int n,m;
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++) scanf("%d",&a[i]);
sort(a+1,a+1+m);
for(int i=1;i<n;i++)
{
for(int i=1;i<=m;i++)
scanf("%d",&b[i]);
sort(b+1,b+1+m);
while(!que.empty()) que.pop();
for(int i=1;i<=m;i++)
{
node t;
t.val=a[i]+b[1];
t.a=i;
t.b=1;
que.push(t);
}
for(int i=1;i<=m;i++)
{
node cur=que.top();
que.pop();
tmp[i] = cur.val;
node nxt;
nxt.val=a[cur.a]+b[cur.b+1];
nxt.a = cur.a;
nxt.b = cur.b+1;
que.push(nxt);
}
for(int i=1;i<=m;i++)
a[i]=tmp[i];
}
for(int i=1;i<=m;i++)
printf("%d%c",a[i],i==m? '\n':' ');
}
return 0;
}