POJ_2442_Sequence(堆的应用)

http://poj.org/problem?id=2442

/*
题意：输入第一行为测试数据，
每一组测试数据有两个数,n、m；n代表接下来输入多少行，m代表每一行
有多少个数；让你计算从n行当中每一行取一个数相加得到的和那么会有n*m
个和。输出这些和当中前m小的。
题意理解之后关键是怎么做，这道题归为堆肯定是用堆来做的。
思路：对两行data1和data2进行操作，对两组数组进行排序
Q[i] = data2[0] + data1[i]
用make_heap对Q建堆。
然后data2[1] + data1[i] (0<=i<=n-1),如果data2[1] + data1[i]
比堆dataq的顶点大,则退出，否则删除堆的顶点，插入data2[1] + data1[i]。
然后是data2[2],...data2[n - 1]
*/

# include <stdio.h>
# include <iostream>
# include <algorithm>
using namespace std;
const int MAXN=2013;
int data1[MAXN],data2[MAXN],Q[MAXN];
int main()
{
    int tests,n,m,i,j,k;
    scanf("%d",&tests);
    while(tests--)
    {
        memset(data1,0,sizeof(data1));
        scanf("%d%d",&n,&m);
        for(i=0;i<m;i++)
            scanf("%d",&data1[i]);
        for(k=1;k<n;k++)
        {
            sort(data1,data1+m);
            memset(data2,0,sizeof(data2));
            
            for(i=0;i<m;i++)  
                scanf("%d", &data2[i]);  
            sort(data2,data2+m);
            
            for(i=0;i<m;i++)
                Q[i]=data1[0]+data2[i];
            make_heap(Q,Q+m);
            for(i=1;i<m;i++)
            {
                for(j=0;j<m;j++)
                {
                    int temp;
                    temp=data1[i]+data2[j];
                    if(temp <= Q[0])
                    {
                        pop_heap(Q,Q+m);
                        Q[m-1]=temp;
                        push_heap(Q,Q+m);
                    }
                }
            }    
            for(int l=0;l<m;l++)
                data1[l]=Q[l];
        }
        sort(data1,data1+m);
        for(i=0;i<m;i++)
        {
            if(i)printf(" ");
            printf("%d",data1[i]);
        }
        printf("\n");
    }
    return 0;
}

 使用优先队列做：

#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;
const int maxn = 2000+10;
int a[maxn], b[maxn];
int main()
{
    int T, n, m;
    int i,j,k;
    scanf("%d", &T);
    while(T--) 
    {
        priority_queue<int> que;       
        scanf("%d %d", &n, &m);        
        for(i=0; i<m; i++)
            scanf("%d", &a[i]);        
        sort(a, a+m);
        for(k=1; k<n; k++)
        {
            for(i=0; i<m; i++)        
                scanf("%d", &b[i]);
            sort(b, b+m);         
            for(i=0; i<m; i++)             
                que.push(a[0] + b[i]);
            for(i=1; i<m; i++)
            {
                for(int j=0; j<m; j++)
                {
                    if(a[i] + b[j] > que.top())
                        break;
                    que.pop();
                    que.push(a[i]+b[j]);
                }
            }          
            for(i=m-1; i>=0; i--) 
            {
                a[i] = que.top();
                que.pop();
            }
        }
        for(i=0; i<m; i++)
        {
            if(i != m-1) printf("%d ", a[i]);
            else printf("%d", a[i]);
        }
        putchar('\n');
    }
    return 0;
}