【JZOJ4678】钱仓

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 longusing 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;}