1. 程式人生 > >POJ 2442 Sequence【堆】

POJ 2442 Sequence【堆】

題目大意:給出一個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;
}