Description
Solution
看似很复杂,实际上我们只要贪心的想,每一个有钱的位置肯定沿顺时针铺下去是最优的。所以我们只要找到合法的起点就可以了。
何为合法?
我们把原序列加倍,从一个起点开始往后取n个数,如果按照上面规则能刚好铺完(由于已经破环,所以不能绕圈铺),那么这就是合法的。
然而合法的起点有许多,取最小值会超时。
我们发现,对于每个合法序列,它们中的钱的mol数都是一样的,所以只是铺的顺序不同,那么贡献也是一样的。
Code
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define N 100001
#define ll long long
using namespace std;
int c[N*2];
int main()
{
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
int n;
cin>>n;
fo(i,1,n) scanf("%d\n",&c[i]),c[i+n]=c[i];
ll ans=0;
int st,mx=0,qz=0,q=1;
fo(i,1,n*2-1)
{
if(qz<0) qz=0,q=i;
qz+=c[i]-1;
if(qz>mx) mx=qz,st=q;
}
int p=st-1;
fo(i,st,st+n-1)
if(c[i])
{
fo(j,1,c[i])
p++,ans+=(p-i)*(p-i);
}
cout<<ans;
}