E. Are You Fired?

Codeforces Round #645 (Div. 2)

题意：
一个长度为n的数组，前(n+1)/2个数随机，剩下的数都为x；
找一个k，满足 任意长度为 k 的 区间（连续）的和 > 0

思路：

1. k > n / 2， 若存在 k < n/2 且满足要求，那么 k * 2 一定满足，小区间都满足了，大区间显然更满足

2. 如果 x >= 0：
   a. 总和 > 0，直接输出n；
   b. 如果总和 < 0，那么一定不存在；因为 k > n / 2，加了剩下所有的x，都 < 0，那么第一个区间无论长度多大，都不可能 > 0

3. 如果 x < 0：
   sum 指 前缀和
   从i 开始 长度为k的区间和 = sum[i + k] - sum[i] > 0，0 <= i <= (n+1)/2
   sum[i + k] = sum[n] - (n - i - k) * x
   sum[k] = sum[n] - (n - k) * x
   合并得：sum[i] - i * x < sum[k]
   即，对于每一个k，当 0 <= i <= n-k时，求一个最大的sum[i] - i * x

   而这个是可以递推的，即 sum[i] - i * x = max(sum[i] - i * x, sum[i-1] - (i - 1) * x)

   最后只要查看是否存在 sum[k] > sum[n-k]，存在即是答案，否则就没有答案

代码：

int n;
ll s[500005];
ll sum[500005];
ll m[500005];
 
void work(){
	cin>>n;
	for(int i=1;i<=(1+n)/2;++i) cin>>s[i];
	ll x; cin>>x;
	for(int i=(1+n)/2+1;i<=n;++i) s[i]=x;
	for(int i=1;i<=n;++i) sum[i]=sum[i-1]+s[i];
 
	if(x>=0){
		cout<<(sum[n]>0? n:-1)<<'\n';
		return;
	}
 
	for(int i=1;i<=(n+1)/2;++i) m[i]=max(m[i-1], sum[i]-i*x);
 
 
	for(int i=(n+1)/2;i<=n;++i) if(m[n-i]<sum[i]){ cout<<i<<'\n'; return;}
	cout<<-1<<endl;
}