CodeForces - 1358E Are You Fired?

Solution

感觉自己写的有点丑。

首先有一个很重要的结论：最后选取的 $k$ 是大于等于 $\lceil \frac{n}{2}\rceil$ 的。原因是如果 $k$ 满足条件，$2\ast k$ 可以由两个长度为 $k$ 的段凑出来。

然后有个挺巧妙的思路：假设在 $x$ 段之前的长度为 $l$ 即 $n-\lfloor \frac{n}{2}\rfloor$。我们先以 $l$ 作为长度，预处理出数组 $sum[i]$ 表示以 $i$ 为起点的和。再令 $pre[i]$ 为 $[1,i]$ 中 $sum$ 数组的最小值。

你会发现，无论长度怎么增长（之前已经证明如果有解，$k$ 一定可以取在 $[l,n]$），$sum$ 数组内部的大小关系都不会变了，因为增加的数都是 $x$ 而且个数相同。

这个时候就可以枚举 $k$，用之前的 $pre$ 增加几个 $x$ 判断是否大于 $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;
}