CF1680D Dog Walking 题解

Description

传送门

Solution

Key Observation

对于任意序列 $a$ 而言，其前缀最大值与前缀最小值的差，即为 $a$ 的各个循环排列的最大前缀和。

例如，对于 a = { − 4 , 2 , 3 , 3 , − 4 } a=\{-4,2,3,3,-4\} a={−4,2,3,3,−4}，其前缀最大值与前缀最小值的差 $4-(-4)=8$ 即为其循环排列 { 2 , 3 , 3 , − 4 , − 4 } \{2,3,3,-4,-4\} {2,3,3,−4,−4} 的前缀 $[1,3]$ 的和。

---

考虑枚举 $a$ 的一个循环排列，那么，我们需要把 $a$ 中的每个 $0$ 替换为某个数使得其最大前缀和最大。

考虑贪心，从左往右扫描 $a$，每次在序列合法的前提下，尝试填入最大的数。具体来说，当我们要替换 $a_i(a_i=0)$ 时，求出 $[i+1,n]$ 的和 $s$ 及其中 $0$ 的数量 $c$，那么 $a[i+1,n]$ 的和就可以为 $[s-ck,s+ck]$ 中的任何一个数，于是就能计算得到填入的最大数了。

时间复杂度 $O(n^2)$。注意开 long long。

Code

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxl=3005;

int read(){
	int s=0,w=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-')  w=-w;ch=getchar();}
	while (ch>='0'&&ch<='9'){s=(s<<1)+(s<<3)+(ch^'0');ch=getchar();}
	return s*w;
}
int n,k,L,R,res,a[maxl],cnt[maxl],suf[maxl];
void Rotate(){
	int orin=a[n];
	for (int i=n;i>=2;i--)  a[i]=a[i-1];
	a[1]=orin;
}

signed main(){
	n=read(),k=read();
	for (int i=1;i<=n;i++){
		a[i]=read();
		if (a[i])  L+=a[i],R+=a[i];
		else L-=k,R+=k;
	}
	if (R<0||L>0)  return puts("-1"),0;
	
	for (int i=1;i<=n;i++){
		int sum=0;
		for (int j=n;j>=1;j--)  suf[j]=suf[j+1]+a[j],cnt[j]=cnt[j+1]+(a[j]==0);
		for (int j=1;j<=n;j++){
			if (a[j])  sum+=a[j];
			else{
				int x=min(k*cnt[j+1]-suf[j+1]-sum,k);
				if (x>=-k)  sum+=min(x,k);
				else break;
			}
			res=max(res,sum);
		}
		Rotate();
	}
	cout<<res+1<<endl;
	return 0;
}