题意:
给你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 ;
}