题意:给你n*m的矩阵,然后每行取一个元素,组成一个包含n个元素的序列,一共有n^m种序列,
让你求出序列和最小的前n个序列的序列和。
先介绍下堆的基本操作:
int arr[N];
make_heap(arr+1,arr+N);//对数组arr建堆,堆里第一个元素为最大值
pop_heap(arr+1,arr+N+1);//将第一个元素与最后一个元素交换
push_heap(arr+1,arr+N+1);//录入元素
sort_heap(arr+1,arr+N+1);//排序
思路:用普通方法超时,用堆做 具体如下:
1.将第一序列读入sum中,排序。
2.将数据读入num向量中排序。
用arr数组存sum[1] + num[i] ( 1<=i<=n)的和
用make_heap对arr建堆。
然后for循环比较sum[1] + num[i] (1<=i<=n)与堆arr的顶点,前者大则退出,否则删除
堆的顶点,插入前者。
3.将arr的数据拷贝到sum中,并对sum按升序排序
4.循环2,3步,直到所有数据读入完毕。
5.打印sum中的数据即为结果。
代码:
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <stdlib.h>
#include <string.h>
#define N 3000
#define INF 10000000
#define eps 10E-9
#define mem(a) memset(a,0,sizeof(a))
#define w(a) while(a)
#define s(a) scanf("%d",&a)
#define ss(a,b) scanf("%d%d",&a,&b)
using namespace std;
int arr[N],num[N],sum[N];
int main()
{
int n,i,j,k,m,t;
s(t);
w(t--)
{
mem(sum);
ss(m,n);
for(i=1; i<=n; i++)
s(sum[i]);
for(i=2; i<=m; i++)
{
mem(num);
sort(sum+1,sum+n+1);
for(j=1; j<=n; j++)
{
s(num[j]);
}
sort(num+1, num+n+1);
for(j=1; j<=n; j++)
arr[j]=sum[1]+num[j];
make_heap(arr+1,arr+n+1);
for(j=2; j<=n; j++)
{
for(k=1; k<=n; k++)
{
int cnt=sum[j]+num[k];
if(cnt>=arr[1])
break;
pop_heap(arr+1,arr+n+1);
arr[n]=cnt;
push_heap(arr+1,arr+n+1);
}
}
for(j=1; j<=n; j++)
sum[j]=arr[j];
}
sort(sum+1,sum+n+1);
for(i=1; i<n; i++)
printf("%d ",sum[i]);
printf("%d\n",sum[i]);
}
return 0;
}