这道题的数据真强大,各种写法不是MLE就是TLE,好不容易换了这种写法终于AC了。
题目大意:
输入m个数集,每个含n个数,求从每个集合取一个数后,按非降序输出前n小的和。
用优先队列做的,注意限制队列元素个数,否则会超时超内存的。
下面是代码:
#include <stdio.h>
#include <queue>
using namespace std;
int n,m,a[4000005],b[4000005];
int cmp(const void *a,const void *b)
{
return *(int *)a-*(int *)b;
}
int main()
{
int t,flat,i,j,sum,k;
scanf("%d",&t);
while(t--)
{
int ai=1,bi=0,x,r,sum,a[2005];
scanf("%d%d",&n,&m);
priority_queue <int ,vector<int >,greater<int> >p;
priority_queue <int ,vector<int >,less<int> >q;
for(i=0; i<m; i++)
{
scanf("%d",&x);
p.push(x);
}
for(i=1; i<n; i++)
{
for(j=0; j<m; j++)
{
scanf("%d",&a[j]);
}
while(!p.empty())
{
sum=p.top();
p.pop();
for(j=0;j<m;j++)
{
if(q.size()==m&&q.top()>sum+a[j])
{
q.pop();
q.push(sum+a[j]);
}
else if(q.size()<m)
{
q.push(sum+a[j]);
}
}
}
while(!q.empty())
{
p.push(q.top());
q.pop();
}
}
flat=0;
for(i=0;i<m;i++)
{
if(flat)
{
printf(" ");
}
printf("%d",p.top());
p.pop();
flat=1;
}
printf("\n");
}
return 0;
}