(優先佇列)P1631 序列合併
阿新 • • 發佈:2018-11-28
https://www.luogu.org/problemnew/show/P1631
有兩個長度都是N的序列A和B,在A和B中各取一個數相加可以得到N^2
個和,求這N^2個和中最小的N個。
Ai <= 1e9, Bi <= 1e9, N <= 1e5.
直接暴力用優先佇列存前n個,如果出現第一個比當前最大值還大的值時退出迴圈保障複雜度
#include<bits/stdc++.h> using namespace std; const int maxn = 1e5 + 10; typedef long long ll; priority_queue< ll, vector<int>, less<int> > q; int a[maxn]; int main() { int n, x; cin >> n; for(int i = 0; i < n; i++) cin >> a[i]; for(int i = 0; i < n; i++) { cin >> x; for(int j = 0; j < n; j++) { ll num = x + a[j]; if(q.size() < n) q.push(num); else { if(num < q.top()) { q.pop(); q.push(num); } else break; } } } int pos = 0; while(!q.empty()) { a[n - pos++ - 1] = q.top(); q.pop(); } for(int i = 0; i < n; i++) cout << a[i] << " "; }