郑州轻工业OJ minval 优先队列

minval

题目描述

有两个长度为N的序列A和B，在A和B中各任取一个数相加可以得到N2个和，求这N2个和中最小的N个。

输入

第一行输入一个正整数N（1<=N<=100000）；

第二行N个整数Ai且Ai<=109；第三行N个整数Bi且Bi<=109。

输出

输出仅一行，包含n个整数，从小到大输出这n个最小的和，相邻数字之间用空格隔开。

样例输入

51 3 2 4 56 3 4 1 7

样例输出

2 3 4 4 5

来源

 优先队列

#include<stdio.h>
#include<queue>
#include<algorithm>
using namespace std;
long long a[100005],b[100005],ans[100005];
int main(){
	int N;
	//priority_queue<long long,vector<long long>,greater<long long> >q;
	priority_queue<long long>q;//默认的从小到大排序 
	scanf("%d",&N);
	for(int i=1;i<=N;i++)
	scanf("%d",&a[i]);
	for(int i=1;i<=N;i++)
	scanf("%d",&b[i]);
	sort(a+1,a+N+1);
	sort(b+1,b+N+1);
	for(int i=1;i<=N;i++)
		q.push(a[1]+b[i]);
	for(int i=2;i<=N;i++){
		for(int j=1;j<=N;j++){
			if(a[i]+b[j]<q.top()){
				q.pop();
				q.push(a[i]+b[j]);
			}
			else
			break;//如果当前值比队列中所有的值大（因为是优先队列，所以只要比一个值大就是比所有值大了）就跳出，
					//而且因为数组是从小到大排好序的，如果当前值都大，那后边的值一定会更大 
		}
	}
	int k=1;
	while(!q.empty()){//因为队列中默认的是从大到小的放的，所以从小到大的输出需要再加一个数组 
		ans[k++]=q.top();
		q.pop();
	}
	for(int i=N;i>=1;i--){
		if(i==1)
		printf("%lld\n",ans[i]);
		else
		printf("%lld ",ans[i]);
	}
	return 0;
}