题目描述
有两个长度都为N的序列A和B,在A和B中各取一个数相加可以得到N2个和,求这N2个和中最小的N个。
输入
第一行一个正整数N(1 <= N <= 100000)。
第二行N个整数Ai,满足Ai <= Ai+1且Ai <= 109
第三行N个整数Bi,满足Bi <= Bi+1且Bi <= 109
输出
输出仅有一行,包含N个整数,从小到大输出这N个最小的和,相邻数字之间用空格隔开。
样例输入 复制
3
2 6 6
1 4 8
样例输出 复制
3 6 7
分析:
求一堆数中最小的N个数,当然可以对全部数进行堆排序或者快排,然后取前N个,但浪费空间,可以维护大小为N的数组或者大顶堆,每次读取一个数与其最大的数比较,保证数组或者堆中都是目前最小的N个数。
要点:
进行循环条件时,切记要注意是否出现循环条件在循环中被改变的情况。
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
int main()
{
int N;
while(cin >> N){
vector<int> A, B;
int e;
for(int i = 0; i < N; i++){
cin >> e;
A.push_back(e);
}
priority_queue<int> q;
for(int i = 0; i < N; i++){
cin >> e;
q.push(e + A[0]);
B.push_back(e);
}
for(int i = 1; i < A.size(); i++){
for(int j = 0; j < B.size(); j++){
int temp = A[i] + B[j];
if(temp < q.top()){
q.pop();
q.push(temp);
}else{
break;
}
}
}
vector<int> re;
for(int i = 0; i < N; i++){
re.push_back(q.top());
q.pop();
}
for(int i = N - 1; i >= 0; i--){
cout << re[i];
if(i == 0) cout << '\n';
else cout << ' ';
}
}
return 0;
}