uva 11997 - K Smallest Sums

题意：有k个整数数组，各包含k个元素。在每个数组中取一个元素加起来，可以得到 ^kk个和。求这些和中最小的k个值（从小到大输出）。

#include<cstdio>
#include<algorithm>
#include<queue>

using namespace std;

struct node
{
	int s,b;
	node(int s,int b):s(s),b(b){}
};
bool operator < (node x,node y)
{
	return x.s>y.s;
}
void f(int *A,int *B,int *C,int n)
{
	int i,b;
	priority_queue<node>q;
	for(i=0;i<n;i++)
	{
		q.push(node(A[i]+B[0],0));
	}
	for(i=0;i<n;i++)
	{
		node p=q.top();q.pop();
		C[i]=p.s;
		b=p.b;
		if(b+1<n) q.push(node(p.s-B[b]+B[b+1],b+1));
	}
}
int main()
{
	int i,j,k,A[760],B[760];
	while(scanf("%d",&k)!=EOF)
	{
		for(i=0;i<k;i++) scanf("%d",&A[i]);
		sort(A,A+k);
		for(i=1;i<k;i++) 
		{
			for(j=0;j<k;j++) scanf("%d",&B[j]);
			sort(B,B+k);
			f(A,B,A,k);
		}
		printf("%d",A[0]);
		for(i=1;i<k;i++)
			printf(" %d",A[i]);
		printf("\n");
	}
	return 0;
}