洛谷 P1631 序列合并 优先队列

https://www.luogu.com.cn/problem/P1631

代码如下:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<queue>
using namespace std;
priority_queue<int>q;
int n,a[100005],b[100005],ans[100005];
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]);
    }
    sort(a+1,a+n+1);
    sort(b+1,b+n+1);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            int x=a[i]+b[j];
            if(q.size()<n){
                q.push(x);
            }
            else{
                if(q.top()>x){
                    q.pop();
                    q.push(x);
                }
                else{
                    break;
                }
            }
        }
    }
    for(int i=n;i>=1;i--){
        ans[i]=q.top();
        q.pop();
    }
    for(int i=1;i<=n;i++){
        printf("%d ",ans[i]);
    }
    return 0;
}
发布了253 篇原创文章 · 获赞 15 · 访问量 7994

猜你喜欢

转载自blog.csdn.net/weixin_44123362/article/details/103895423