【JZOJ 4678】钱仓

Description

$n\leq100000$
PS：题目好坑，描述不清
题意的目标：使得所有位置都为1.

Analysis

首先把环破坏成链。
枚举断开位置，然后贪心地运钱。
什么样的贪心策略是对的呢？
可以通过手算或者直接证明($x^2+y^2\leq (x+y)^2$)发现，一个位置上若有0肯定用其左边的第一个非0的位置搬过来。如果搬过来那个位置变成0也无妨，那个位置再借用它左边的就行了。这样一定是最优的。
但是，可能有一些位置断开后0太多了，钱不够用了QAQ。
所以要枚举到一个合法的位置。
对于多个位置都合法，应选取最小的那个位置。
但是这样是$O(n^2)$的。
事实上，只要某个位置合法，它与其他合法位置的值是一样的（挪移一下）。
所以只要找出一个合法位置并输出即可。
这样可以预处理，然后就变成$O(n)$了。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
using namespace std;
typedef long long ll;
const int N=200010;
int a[N],sm[N];
int main()
{
    int n,st=1;
    scanf("%d",&n);
    fo(i,1,n) scanf("%d",&a[i]),a[n+i]=a[i];
    fo(i,1,n+n-1) sm[i]=sm[i-1]+a[i]-1;
    fo(i,n,n+n-1)
        if(sm[i]-sm[i-n]){st=i;break;}
    for(int i=1;i<=n+n-1;i++)
    {
        while(sm[i]-sm[st-1]<0) st=++i;
        if(i-st+1==n) break;
    }
    int j=st+n-1;
    ll ans=0;
    fd(i,st+n-1,st)
        if(!a[i])
        {
            while(a[j]==0 || j>i) j--;
            a[i]++,a[j]--,ans+=(j-i)*(j-i);
        }
    printf("%lld",ans);
    return 0;
}