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] << " ";
}