序列-------------------------------思维(二叉堆)

在这里插入图片描述
在这里插入图片描述
解析:
两步走:
第一步: 因为存在很多行,不能暴力合并匹配。我们可以先求出第一第二行合并后的m个最小值。然后用求出来的这个数组再去和下面的数组合并 一共会进行m-1次操作

第二步: 怎么选出m个最小值呢,当然n2 做法是不可取 。所以我们利用分组+二叉堆可以O(mlogn) 求出
假设现在只有2行
a1,a2,a3…am
b1,b2,b3…bm
我们给a数组排序,然后分组
a1+b1,a2+b1,a3+b1…am+b1
a1+b2,a2+b2,a3+b2…am+b2

a1+bm,a2+bm,a3+bm…am+bm

这样就分组完成了。
对于每一组的内部都是已经排序。因为a数组升序,对于每个数都加上一个值,是不影响单调性的。
所以每个分组都已经排好序。

那么找出最小值,就是每个分组的开头
还有一个细节就是:
假设S=a1+b2, 我把第二组的开头取走了。取下一个最小值的时候就是a2+b2 和其他开头作比较了。

找最小值用二叉堆维护即可

所以总的时间复杂度 O(nmlog(m))

#include<bits/stdc++.h>
using namespace std;
#define x first
#define y second
const int   N=1e5+100;
typedef pair<int,int> PII;
int t;
int n,m;
int a[N],b[N],c[N];
void merge()
{
    priority_queue<PII,vector<PII>,greater<PII > > q;
    for(int i=0;i<m;i++) q.push({a[0]+b[i],0});
    for(int i=0;i<m;i++)
    {
        auto t=q.top();q.pop();
        c[i]=t.x;int pos=t.y;
        q.push({t.x-a[pos]+a[pos+1],pos+1});
    }
    for(int i=0;i<m;i++) a[i]=c[i];
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&m);
        for(int i=0;i<m;i++) scanf("%d",&a[i]);
        sort(a,a+m);
        for(int i=0;i<n-1;i++)
        {
            for(int j=0;j<m;j++) scanf("%d",&b[j]);
            merge();
        }
        for(int i=0;i<m;i++) cout<<a[i]<<" ";
        cout<<endl;
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值