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