Uva-11997-K Smallest Sums

题目要求是从k个数组中各选一个数组成的和中最小的k个。

用优先队列做，注意合并的时候可以做一些适当的优化操作，因为最开始已经对每个数组进行了排序，所以每一次更新只需要从当前优先队列中取出第一个数，然后进行更新操作即可。更新为删除当前相加的值，而加上当前值后面的一个值。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
#define MAX 751
using namespace std;
struct node
{
    int sum;
    int last;
    node(int sum,int last):sum(sum),last(last){}
    bool operator < (const node& a)const
    {
	return sum>a.sum;
    }
};
int a[MAX][MAX],k;
void merge(int *a,int *b)
{
    priority_queue<node> pq;
    for(int i=0;i<k;i++)
	pq.push(node(a[i]+b[0],0));
    for(int i=1;i<k;i++)
    {
	node ita=pq.top();
	pq.pop();
	a[i-1]=ita.sum;
	ita.sum=ita.sum-b[ita.last]+b[++ita.last];
	pq.push(ita);
    }
    a[k-1]=pq.top().sum;
}
int main()
{
    while(scanf("%d",&k)!=EOF)
    {
	for(int i=0;i<k;i++)
	{
	    for(int j=0;j<k;j++)
		scanf("%d",&a[i][j]);
	    sort(a[i],a[i]+k);
	}
	for(int i=1;i<k;i++)
	    merge(a[0],a[i]);
	for(int i=0;i<k;i++)
	    printf("%d%c",a[0][i],i==k-1?'\n':' ');
    }
    return 0;
}