二叉堆——序列（模板）

传送门呢:146. 序列 - AcWing题库

思路：每次处理两个序列，求出这两个序列的前n小的数组成新的序列，一共合并m-1次，第一组序列要先排序，后序因为小根堆不用。

代码：

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
typedef pair<int,int>PII;
const int N=2010;
int n,m;
int a[N],b[N],c[N];
void merge()
{
    priority_queue<PII,vector<PII>,greater<PII>>heap;
    for(int i=0;i<n;i++) heap.push({a[0]+b[i],0});

    for(int i=0;i<n;i++)
    {
        auto t=heap.top();
        heap.pop();
        int s=t.first ,p=t.second;
        c[i]=s;
        heap.push({s-a[p]+a[p+1],p+1});
    }
    for(int i=0;i<n;i++)a[i]=c[i];
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&m,&n);
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);

        sort(a,a+n);

        for(int i=0;i<m-1;i++)
        {
            for(int j=0;j<n;j++) scanf("%d",&b[j]);
                merge();
        }

        for(int i=0;i<n;i++)printf("%d ",a[i]);
        puts("");
    }
        return 0;
}

错误代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
typedef pair<int,int>PII;
const int N=1e4+10;
bool cmp(PII a,PII b)
{
    return a.second<b.second;
}
int cnt[2010];
int d[2010][1010];
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
      //  memset(d,0x3f,sizeof d);
        int n,m;
        for(int i=1;i<=n;i++)
            cnt[i]=1;
        cin>>m>>n;
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=n;j++)
                scanf("%d",&d[i][j]);
            sort(d[i]+1,d[i]+m+1);
        }
        priority_queue<PII,vector<PII>,less<PII>>heap;
        int sum=0;
        for(int i=1;i<=m;i++)
        {
        heap.push({d[i][1],i});
        sum+=d[i][1];
        }
        int k=n;

        while(k--)
        {
            int p=heap.top().second;
            cnt[p]++;
            sum-=heap.top().first;
            heap.pop();
            heap.push({d[p][cnt[p]],p});
            sum+=d[p][cnt[p]];
            cout<<sum<<' ';
        }
        cout<<endl;
    }
        return 0;
}