POJ 2442 Sequence【堆】
阿新 • • 發佈:2018-12-24
題目大意:給出一個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; }