ACwing 146. 序列

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>

using namespace std;

const int N=2010;
int a[N],b[N],c[N];
int m,n;
typedef pair<int,int> pii;

void merge()
{
  
  priority_queue<pii,vector<pii>,greater<pii>>heap;//创建最小根堆
  for(int i=0;i<n;i++)heap.push({a[0]+b[i],0});//参数2记录的是行下标，参数1是值；
  
  for(int i=0;i<n;i++)
  {
    auto t=heap.top();//取出最小值；
    heap.pop();
    int s=t.first,p=t.second;//s是两数和，t是数组a的下标
    c[i]=s;
    heap.push({s-a[p]+a[p+1],p+1});//此时把同行除去pop掉的，此时最大的数加入堆；
    
  }
  for(int i=0;i<n;i++)a[i]=c[i];//此时a[i]的值也是升序的，因为pop顺序是按集合升序pop的；
}



int main()
{
  int T;
  //scanf("%d",&T);
  cin>>T;
  while(T--)
  {
    cin>>m>>n;
    //scanf("%d%d",&m,&n);
    for(int i=0;i<n;i++)cin>>a[i];//scanf("%d",&a[i]);
    sort(a,a+n);
    for(int i=0;i<m-1;i++)
    {
      for(int j=0;j<n;j++)cin>>b[j];//scanf("%d",&b[j]);
      merge();
    }
    for(int i=0;i<n;i++)cout<<a[i]<<" ";//printf("%d ",a[i]);
    cout<<endl;
    //puts("");
  }
  
  return 0;
}