POJ 2442 Sequence(堆的应用)

poj 2442

题目大意

给出m行n列,每行选一个数组成一个和，有$n^m$个结果，问前n小的和是哪些.

分析

【超内存】一开始想的思路是以一个和作为元素用优先队列，感觉思路还是比较巧妙，有点类似于dijkstra。先把每一行排序，每行的第一个元素组成的和一定是最小的，通过改变最小的和中某个数（将它右移一位）可以得到n个新的数，将这些数加入优先队列，每次选择优先队列里最小的和来进行上述的操作。这样的一个问题是会爆内存，因为每取出一个和，又会加入n个进去，所以会需要n*n的空间。
限制优先队列的个数还是应该还是可以过的。
【改后思路】看博客后发现和别人的思路不一样，没有用结构体，大体思路是这样的：把前k行每行一个数组成的前n小的和保存下来,在向前k+1递推.
这里用到了STL中的堆

代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<algorithm>
using namespace std;
const int INF=999999999;
int T;
int m,n;
int a[105][2005];
int sum[2005];
bool cmp(int a,int b)
{
      return a<b;
}
void Work()
{
      int temp[2005];
      for(int i=1;i<=n;i++)
            sum[i]=a[1][i];
      make_heap(&sum[1],&sum[n+1],cmp);

      for(int i=2;i<=m;i++)
      {
            for(int j=1;j<=n;j++)
            {
                temp[j]=sum[j]+a[i][1];
            }
            make_heap(&temp[1],&temp[n+1],cmp);
            sort(sum+1,sum+1+n);
            for(int j=2;j<=n;j++)
            {
                  for(int k=1;k<=n;k++)
                  {
                        if(sum[k]+a[i][j]<temp[1])
                        {
                            pop_heap(&temp[1],&temp[n+1]);
                            temp[n]=sum[k]+a[i][j];
                            push_heap(&temp[1],&temp[n+1]);
                        }
                        else break;
                  }
            }
            for(int j=1;j<=n;j++)sum[j]=temp[j];
      }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
          scanf("%d%d",&m,&n);
          for(int i=1;i<=m;i++)
          {
                for(int j=1;j<=n;j++)
                {
                     scanf("%d",&a[i][j]);
                }
                sort(a[i]+1,a[i]+n+1);
          }
          Work();
          sort(sum+1,sum+n+1);
          for(int i=1;i<n;i++)cout<<sum[i]<<" ";
          cout<<sum[n]<<endl;
    }
}
/*
1
2 3
3 2 1
2 2 3
*/