[洛谷P1631]序列合併/[codevs1245]最小的N個和
阿新 • • 發佈:2019-01-10
描述
有兩個長度都是N的序列A和B,在A和B中各取一個數相加可以得到N^2個和,求這N^2個和中最小的N個。
輸入輸出格式
輸入格式:
第一行一個正整數N;
第二行N個整數Ai,滿足Ai<=Ai+1且Ai<=10^9;
第三行N個整數Bi, 滿足Bi<=Bi+1且Bi<=10^9.
【資料規模】
對於50%的資料中,滿足1<=N<=1000;
對於100%的資料中,滿足1<=N<=100000。
輸出格式:
輸出僅一行,包含N個整數,從小到大輸出這N個最小的和,相鄰數字之間用空格隔開。
輸入樣例:
3
2 6 6
1 4 8
輸出樣例:
3 6 7
題解:想了很久才懂。。。ORZ。。
給兩個陣列從大到小排個序。然後a[1]顯然是陣列a裡面最小的元素。
先往小根堆裡面丟a[1]+b[i](1<=i<=n)
再拿一個g陣列,g[i]表示當前堆內a[k]+b[i]中最大的k。很顯然a[g[i]]+b[i]是b[i]當前堆內所對應的最大和,那麼接下來我們只需要把堆中最小的元素取出然後輸出,比如這個元素是a[g[x]]+b[x],那麼我們還要往堆內加入a[g[x]+1]+b[x],並且把g[x]更新成g[x]+1。把取出最大值操作和更新操作迴圈著做N次就好啦。
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<queue>
#include<algorithm>
#define pa pair<long long,int>
#define LiangJiaJun main
using namespace std;
priority_queue<pa,vector<pa>,greater<pa> >q;
int n,g[100004];
long long a[100004],b[100004];
int LiangJiaJun(){
scanf ("%d",&n);
for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
for(int i=1;i<=n;i++)scanf("%lld",&b[i]);
sort(a+1,a+n+1);sort(b+1,b+n+1);
for(int i=1;i<=n;i++){
q.push(make_pair(a[1]+b[i],i));
g[i]=1;
}
int p=n;
while(p--){
pa x=q.top();q.pop();
printf("%lld ",x.first);
g[x.second]++;
q.push(make_pair(a[g[x.second]]+b[x.second],x.second));
}
printf("\n");
return 0;
}