POJ 2442 Sequence【堆】

题目链接：http://poj.org/problem?id=2442

题目大意：给出一个m*n的矩阵，从每一行中取出一个数相加。能得到n^m个不同的结果。要求输出当中前n项。

建立一个以n元数组为底层数组的堆，在这里，利用stl中的make_heap，pop_heap。push_heap等函数解决。

1.将第一组数据输入arr1数组。升序排序。

2.将接下来的数据输入到arr2数组中。而且heap[i]=arr1[0]+arr2[0...n-1]。make_heap(heap,heap+n).

3.arr1数组从1到n-1，比較temp=arr1[i]+arr2[0...n-1]与堆顶的元素。假设temp比較小，则将堆顶元素pop。加入temp到heap；否则跳出循环。

4.将heap中的元素所有赋值给arr1数组，升序排序。反复2,3两步，直到所有数据所有处理完。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#define M 111
#define N 2111
using namespace std;
int arr1[N],arr2[N],heap[N];
int m,n;
int main()
{
    int t;
    bool s=true;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&m,&n);
        for(int i=0;i<n;i++)
            scanf("%d",&arr1[i]);
        sort(arr1,arr1+n);
        for(int i=1;i<m;i++)
        {
            for(int j=0;j<n;j++)
                scanf("%d",&arr2[j]);
            sort(arr2,arr2+n);
            for(int j=0;j<n;j++)
                heap[j]=arr1[0]+arr2[j];
            make_heap(heap,heap+n);
            for(int j=1;j<n;j++)
            {
                for(int k=0;k<n;k++)
                {
                    int temp=arr1[j]+arr2[k];
                    if(temp<heap[0])
                    {
                        pop_heap(heap,heap+n);//将heap里的最大值放在最后，而且又一次修正heap
                        heap[n-1]=temp;
                        push_heap(heap,heap+n);
                        s=false;
                    }
                    else
                        break;
                }
                if(s)
                    break;
            }
            for(int j=0;j<n;j++)
                arr1[j]=heap[j];
            sort(arr1,arr1+n);
        }
        for(int i=0;i<n;i++)
            printf("%d ",arr1[i]);
        printf("\n");
    }
    return 0;
}