第一次寫堆,估計也可以用優先隊列做,給一個m*n矩陣,每行任取一個數出來,求最小n 數和。從第一行到第i行,用一個堆去存儲前i行,最小的n個i數和。
#include <iostream>
#include <cstring>
#include <string>
#include <queue>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[2005],b[2005],sum[2005];
int main()
{
int t,m,n;
scanf("%d",&t);
while(t--)
{
cin>>m>>n;
for(int i=0;i<n;i++)scanf("%d",&a[i]);
sort(a,a+n);
for(int i=1;i<m;i++)
{
for(int i=0;i<n;i++)
scanf("%d",&b[i]);
sort(b,b+n);
for(int i=0;i<n;i++)sum[i]=a[0]+b[i];
make_heap(sum,sum+n);
for(int j=1;j<n;j++)
for(int k=0;k<n;k++)
{
int temp=a[j]+b[k];
if(temp>sum[0])break;
pop_heap(sum,sum+n);
sum[n-1]=temp;
push_heap(sum,sum+n);
}
for(int i=0;i<n;i++)
a[i]=sum[i];
}
sort(a,a+n);
for(int i=0;i<n-1;i++)
printf("%d ",a[i]);
printf("%d\n",a[n-1]);
}
return 0;
}