刷题记录-luoguP1631 序列合并

来源:互联网 发布:人工智能取代人类辩论 编辑:程序博客网 时间:2024/06/09 21:57

把a数组分成n部分,对于每个a[i]都看成加上整个b数组。

比如样例:

32 6 61 4 8
可看成

2-> 1 4 8

6-> 1 4 8

6-> 1 4 8

设c[i]为当前a[i]未处理的最小小标,所有的a[i]+b[c[i]]即可构成组合的最小序列,

在其中用堆在序列中可以找到最小的,这样就是n*log n了

讲的不是很清楚,详见代码

***********************************

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define pii pair<int,int>
#define MAXN 100005
using namespace std;
int n;
int a[MAXN],b[MAXN];
int c[MAXN];
priority_queue<pii,vector<pii >,greater<pii > > q;  
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&b[i]);
    }
    for(int i=1;i<=n;i++){
        c[i]=1;
        q.push(make_pair(a[i]+b[c[i]],i));
    }
    for(int i=1;i<=n;i++){
        pii p=q.top(); q.pop();
        int x=p.second;
        c[x]++; q.push(make_pair(a[x]+b[c[x]],x));
        printf("%d ",p.first);
    }
    return 0;
}

原创粉丝点击