您的位置:首页 > 其它

刷题记录-luoguP1631 序列合并

2017-07-12 22:40 218 查看
把a数组分成n部分,对于每个a[i]都看成加上整个b数组。

比如样例:

3
2 6 6
1 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;

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: