POJ 2442 Sequence

来源：互联网发布：大数据入门经典书籍编辑：程序博客网时间：2024/05/24 03:19

题目大意: 有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>

0 0