(优先队列)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] << " ";
}

猜你喜欢

转载自blog.csdn.net/weixin_40588429/article/details/84349280