POJ 2442 Squence (STL heap)

时间:2023-12-24 15:23:37

题意:

  给你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)中的序列进行排序。它假设这个序列是有效堆。(当然
,经过排序之后就不是一个有效堆了)

这样,这个题的实现就很简单了:

 #include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
int seq2[],seq1[],seqn[];
int main()
{
int t;
int m,n;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&m,&n);
for(int i=;i<n;i++)
scanf("%d",&seq1[i]);
sort(seq1,seq1+n);
for(int i=;i<m;i++)
{
memset(seq2,,sizeof(seq2));
for(int j=;j<n;j++)
{
scanf("%d",&seq2[j]);
}
sort(seq2,seq2+n);
for(int j=;j<n;j++)
{
seqn[j]=seq1[]+seq2[j];
}
make_heap(seqn,seqn+n);
for(int j=;j<n;j++)
{
for(int k=;k<n;k++)
{
int tmp=seq1[j]+seq2[k];
if(tmp>=seqn[])
break;
pop_heap(seqn,seqn+n);
seqn[n-]=tmp;
push_heap(seqn,seqn+n);
}
}
for(int j=;j<n;j++)
{
seq1[j]=seqn[j];
}
sort(seq1,seq1+n);
}
for(int i=;i<n-;i++)
{
printf("%d ",seq1[i]);
}
printf("%d\n",seq1[n-]);
}
return ;
}