Description
在宇宙中存在着一种神秘的暗物质魔法球。这种魔法球分为阴球和阳球,当两种球合并时就会释放出巨大的能量。 每个球都有一个魔法值。现在给定你n对魔法球,请你找出所有的组合中魔法值最大的n种。 ( 组合意思是指一个阴球和一个阳球结合,魔法值为两球之和,每个魔法球可以使用多次。)
Input
单组输入。 第一行为两个整数n(1<=n<=1e5) 第二行为n个整数,代表n个阳球的魔法值,每个整数值范围(1-1e7)。 第三行为n个整数,代表n个阴球的魔法值,每个整数值范围(1-1e7)。
Output
输出1行,为n个由空格分割的整数,即为最后所求答案,最后一个整数后无空格。
Input
5
1 2 5 3 4
1 2 3 4 5
Output
10 9 9 8 8
用小顶堆或者优先队列来保存答案,样例代码用的小顶堆
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
int a[MAXN], b[MAXN], ans[MAXN];
int n;
void sift_down(int i) {
int t;
while (i <= n / 2) {
if (ans[i] > ans[i * 2])
t = i * 2;
else
t = i;
if (i * 2 + 1 <= n && ans[t] > ans[i * 2 + 1]) t = i * 2 + 1;
if (i != t) {
int x = ans[i];
ans[i] = ans[t];
ans[t] = x;
i = t;
} else {
break;
}
}
}
int main() {
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++) {
ans[i] = a[i] + b[n];
}
for (int i = n; i >= 1; i--) {
int f = 1;
for (int j = n - 1; j >= 1; j--) {
if (a[i] + b[j] > ans[1]) {
f = 0;
ans[1] = a[i] + b[j];
sift_down(1);
}else{
//随着j的减少,b[j]会减小,a[i]+b[j]会更小
break;
}
}
if (f == 1) break;
}
sort(ans + 1, ans + n + 1);
for (int i = n; i > 1; i--) printf("%d ", ans[i]);
printf("%d\n", ans[1]);
return 0;
}