poj2442(堆)

#include<cstdio>
#include<algorithm>
#include<iostream>
#define Max 2008
using namespace std;
int sum[Max],data[Max],Q[Max];
int main()
{
    int i,j,k,t,m,n,temp;
    scanf("%d",&t);
    while(t--)
    {
        memset(sum,0,sizeof(sum));
        scanf("%d %d",&m,&n);
        for(i=0;i<n;i++)
            scanf("%d",&sum[i]);
        for(i=1;i<m;i++)
        {
            memset(data,0,sizeof(data));
            sort(sum,sum+n);
            for(j=0;j<n;j++)
                scanf("%d",&data[j]);
            sort(data,data+n);
            for(j=0;j<n;j++)
                Q[j]=sum[0]+data[j];
            make_heap(Q,Q+n);
            for(j=1;j<n;j++)
            {
                for(k=0;k<n;k++)
                {
                    temp=sum[j]+data[k];
                    if(temp>=Q[0])
                        break;
                    pop_heap(Q,Q+n);
                    Q[n-1]=temp;
                    push_heap(Q,Q+n);
                }
            }
            for(j=0;j<n;j++)
                    sum[j]=Q[j];
        }
        sort(sum,sum+n);
        for(j=0;j<n-1;j++)
            printf("%d ",sum[j]);
        printf("%d\n",sum[n-1]);
    }return 0;
}

给你n*m的矩阵，然后每行取一个元素，组成一个包含n个元素的序列，一共有n^m种序列，

让你求出序列和最小的前n个序列的序列和。

思路  ：

来自下面的博客

http://www.cnblogs.com/vip-wy/archive/2012/02/15/2353064.html

解题步骤：

1.将第一序列读入data1向量中，并按升序排序。

2.将数据读入data2向量中，并按升序排序。

    将data2[0] + data1[i] ( 0<=i<=n-1)读入dataq向量中

    用make_heap对dataq建堆。

    然后data2[1] + data1[i] (0<=i<=n-1),如果data2[1] + data1[i]比堆dataq的顶点大,则退出，否则删除

    堆的顶点，插入data2[1] + data1[i]。然后是data2[2],...data2[n - 1]

 3.将dataq的数据拷贝到data1中，并对data1按升序排序

 4.循环2,3步，直到所有数据读入完毕。

 5.打印data1中的数据即为结果。