循环卷积
题目链接:ybtoj高效进阶 21271
题目大意
给你两个数组,要你求出它们的 max+ 循环卷积。
其实就是如果两个数组的第 i 个和第 j 个加起来,就会放在循环卷积 i+j 模 n 的位置,然后一个位置如果放了多个数就保留最大值。
思路
考虑它 n 虽然很大,但是你看到每个数组里面的数的和都是非负数而且和不超过 5000 5000 5000,这代表很多都是 0 0 0,不是 0 0 0 的数不会超过 5000 5000 5000。
那就好办了,你其实就只需要找不是
0
0
0 的数暴力加来看即可。
当然也有可能会跟
0
0
0 匹配会优(可能只有它一个是正数,其它都是
0
0
0),那这种情况下,我们可以视为把其他数都变成
0
0
0(反正如果不是
0
0
0,后面自然会刷掉这个解),那答案数组的每个位置都可以是由
x
+
0
x+0
x+0 组成,判一下即可。
代码
#include<cstdio>
#include<iostream>
using namespace std;
int n, a[200001], b[200001];
int apl[5001], bpl[5001];
int ans[200001], minn;
int main() {
// freopen("conv.in", "r", stdin);
// freopen("conv.out", "w", stdout);
scanf("%d", &n);
for (int i = 0; i < n; i++) {//找到不是 0 的数组
scanf("%d", &a[i]);
if (!a[i]) continue;
apl[++apl[0]] = i; minn = max(minn, a[i]);//可能除了它都是 0,所以可能跟 0 相加
}
for (int i = 0; i < n; i++) {
scanf("%d", &b[i]);
if (!b[i]) continue;
bpl[++bpl[0]] = i; minn = max(minn, b[i]);
}
for (int i = 1; i <= apl[0]; i++)//暴力
for (int j = 1; j <= bpl[0]; j++) {
int J = apl[i], I = (bpl[j] + J) % n;
// (I - J + n) % n = j
ans[I] = max(ans[I], a[apl[i]] + b[bpl[j]]);
}
for (int i = 0; i < n; i++)
printf("%d ", max(minn, ans[i]));
return 0;
}