Solution
感觉自己写的有点丑。
首先有一个很重要的结论:最后选取的 k k k 是大于等于 ⌈ n 2 ⌉ \lceil \frac{n}{2} \rceil ⌈2n⌉ 的。原因是如果 k k k 满足条件, 2 ∗ k 2*k 2∗k 可以由两个长度为 k k k 的段凑出来。
然后有个挺巧妙的思路:假设在 x x x 段之前的长度为 l l l 即 n − ⌊ n 2 ⌋ n-\lfloor \frac{n}{2} \rfloor n−⌊2n⌋。我们先以 l l l 作为长度,预处理出数组 s u m [ i ] sum[i] sum[i] 表示以 i i i 为起点的和。再令 p r e [ i ] pre[i] pre[i] 为 [ 1 , i ] [1,i] [1,i] 中 s u m sum sum 数组的最小值。
你会发现,无论长度怎么增长(之前已经证明如果有解, k k k 一定可以取在 [ l , n ] [l,n] [l,n]), s u m sum sum 数组内部的大小关系都不会变了,因为增加的数都是 x x x 而且个数相同。
这个时候就可以枚举 k k k,用之前的 p r e pre pre 增加几个 x x x 判断是否大于 0 0 0,如果可行就输出。
Code
#include <cstdio>
#include <iostream>
using namespace std;
typedef long long ll;
const int N = 5e5 + 5;
int n, a[N];
ll sum[N];
int read() {
int x = 0, f = 1; char s;
while((s = getchar()) > '9' || s < '0') if(s == '-') f = -1;
while(s >= '0' && s <= '9') x = (x << 1) + (x << 3) + (s ^ 48), s = getchar();
return x * f;
}
int main() {
n = read();
for(int i = 1; i <= n - (n >> 1); ++ i) a[i] = read();
a[n - (n >> 1) + 1] = read();
for(int i = n - (n >> 1) + 2; i <= n; ++ i) a[i] = a[i - 1];
if(n == 1) return puts(a[1] > 0 ? "1" : "-1"), 0;
for(int i = 1; i <= (n >> 1) + 1; ++ i) sum[1] += a[i];
for(int i = 2; i <= n - (n >> 1); ++ i) sum[i] = sum[i - 1] - a[i - 1] + a[i + (n >> 1)];
for(int i = 2; i <= n - (n >> 1); ++ i) sum[i] = min(sum[i], sum[i - 1]);
for(int i = (n >> 1) + 1; i <= n; ++ i)
if(sum[n - i + 1] - 1ll * ((n >> 1) + 1 - i) * a[n] > 0) return printf("%d\n", i), 0;
puts("-1");
return 0;
}