POJ 2442 Sequence

题目大意:   有m个序列，每个序列有n个数，从每个数列中取出一个数构成一个有m个元素的序列(共有n^m种).,  求这n^m种可能中从小到大输出的前n个sum.(sum指每种情况下各自元素总和)

解题思路:先求前k行的前n个最小sum(利用优先队列进行维护)，再求第k+1行，以此类推,求出m个序列前n个最小sum.

<span style="font-size:14px;">#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#include<cstdio>
using namespace std;
priority_queue<int>que;//最大堆优先队列 
int a[1005][2005];
int sum[2005];
int main()
{
	int T;
	scanf("%d", &T);
	while(T--)
	{
		int i, j,k, m, n;
		scanf("%d%d", &m, &n);
		memset(a,0, sizeof(a));
		memset(sum, 0, sizeof(sum));
		for(i=1; i<=m; i++)
		{
			for(j=1; j<=n; j++)
			{
				scanf("%d", &a[i][j]);
			}
		}
		for(i=1; i<=n; i++)	que.push(a[1][i]);//将第一行的元素入队 
		for(i=2; i<=m; i++)
		{
			sort(a[i]+1, a[i]+n+1);//排队后  从小到大 
			//sum数组储存  以便后边的比较 
			for(j=1; j<=n; j++)
			{
				sum[j]=que.top();
				que.pop();
			}
			//取第i行的第一个元素与前i-1行所取出的前n个最小sum相加 
			for(j=1; j<=n; j++)
			{
				que.push(a[i][1]+sum[j]);
			}
			//  先确定1 2行的前n个最小sum  然后再确定1 2 3行的前n个最小sum  以此类推. 
			for(j=2; j<=n; j++) 
			{
				for(k=n; k>=1; k--)
				{
					if(a[i][j]+sum[k]<que.top())//进行比较,  始终维护堆的前n个 
					{
						que.pop();
						que.push(a[i][j]+sum[k]);
					}
					else break;
				}
			}
		}
		for(i=1; i<=n; i++)
		{
			sum[i]=que.top();
			que.pop();
		}
		for(i=n; i>=1; i--)
		{
			if(i!=n) printf(" ");
			printf("%d",sum[i]);
			if(i==1) printf("\n");
		}
	}
	return 0;
}</span>