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 }

 

posted @ 2015-01-28 17:43  PJQOOO  阅读(264)  评论(0编辑  收藏  举报