1. 程式人生 > >[洛谷P1631]序列合併/[codevs1245]最小的N個和

[洛谷P1631]序列合併/[codevs1245]最小的N個和

描述
有兩個長度都是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; }