首页 > 代码库 > POJ 2442 Squence (STL heap)
POJ 2442 Squence (STL heap)
题意：
给你n*m的矩阵，然后每行取一个元素，组成一个包含n个元素的序列，一共有n^m种序列，
让你求出序列和最小的前n个序列的序列和。
解题思路：
1.将第一序列读入seq1向量中，并按升序排序。
2.将数据读入seq2向量中，并按升序排序。
将seq2[0] +seq1[i] ( 0<=i<=n-1)读入seqn向量中
用make_heap对seqn建堆。
然后seq2[1] + seq1[i] (0<=i<=n-1),如果seq2[1] +seq1[i]比堆seqn的顶点大,则退出，否则删除
堆的顶点，插入seq2[1] +seq1[i]。然后是seq2[2],..seq2[n - 1]
3.将seqn的数据拷贝到seq1中，并对seq1按升序排序
4.循环2,3步，直到所有数据读入完毕。
5.打印seq1中的数据即为结果。
下面给出关于STL中heap的一些用法：
STL里面的堆操作一般用到的只有4个。
他们就是：
make_heap();、pop_heap();、push_heap();、sort_heap();
他们的头函数是algorithm
首先是make_heap();
他的函数原型是：
void make_heap(first_pointer,end_pointer,compare_function);
一个参数是数组或向量的头指针，第二个向量是尾指针。第三个参数是比较函数的名字
。在缺省的时候，默认是大跟堆。（下面的参数都一样就不解释了）
作用：把这一段的数组或向量做成一个堆的结构。范围是(first,last)
然后是pop_heap();
它的函数原型是：
void pop_heap(first_pointer,end_pointer,compare_function);
作用：pop_heap()不是真的把最大（最小）的元素从堆中弹出来。而是重新排序堆。它
把first和last交换，然后将[first,last-1)的数据再做成一个堆。
接着是push_heap()
void pushheap(first_pointer,end_pointer,compare_function);
作用：push_heap()假设由[first,last-1)是一个有效的堆，然后，再把堆中的新元素加
进来，做成一个堆。
最后是sort_heap()
void sort_heap(first_pointer,end_pointer,compare_function);
作用是sort_heap对[first,last)中的序列进行排序。它假设这个序列是有效堆。（当然
，经过排序之后就不是一个有效堆了）
这样，这个题的实现就很简单了：
1 #include<stdio.h> 2 #include<string.h> 3 #include<algorithm> 4 #include<iostream> 5 using namespace std; 6 int seq2[2001],seq1[2002],seqn[2002]; 7 int main() 8 { 9 int t;10 int m,n;11 scanf("%d",&t);12 while(t--)13 {14 scanf("%d%d",&m,&n);15 for(int i=0;i<n;i++)16 scanf("%d",&seq1[i]);17 sort(seq1,seq1+n);18 for(int i=1;i<m;i++)19 {20 memset(seq2,0,sizeof(seq2));21 for(int j=0;j<n;j++)22 {23 scanf("%d",&seq2[j]);24 }25 sort(seq2,seq2+n);26 for(int j=0;j<n;j++)27 {28 seqn[j]=seq1[0]+seq2[j];29 }30 make_heap(seqn,seqn+n);31 for(int j=1;j<n;j++)32 {33 for(int k=0;k<n;k++)34 {35 int tmp=seq1[j]+seq2[k];36 if(tmp>=seqn[0])37 break;38 pop_heap(seqn,seqn+n);39 seqn[n-1]=tmp;40 push_heap(seqn,seqn+n);41 }42 }43 for(int j=0;j<n;j++)44 {45 seq1[j]=seqn[j];46 }47 sort(seq1,seq1+n);48 }49 for(int i=0;i<n-1;i++)50 {51 printf("%d ",seq1[i]);52 }53 printf("%d\n",seq1[n-1]);54 }55 return 0;56 }
POJ 2442 Squence (STL heap)