1. 程式人生 > >(優先佇列)P1631 序列合併

(優先佇列)P1631 序列合併

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