【NOIP2016提高A组8.11】钱仓

Description

Input

第一行n
接下来n行表示每个钱仓中的钱数

Output

答案

Sample Input

10
1
0
0
2
0
0
1
2
2
2

Sample Output

33

Data Constraint

50% n<=5000
100% n<=200000

Solution

50%可以用费用流，看似是裸题。
100%却变成了贪心暴力等
因为$x^2+y^2<(x+y)^2$
所以一个点肯定是往右走到最近空的位置，用两个指针维护
如果这样做，那么易证凡是合法的做法，代价都一样
那么判断合法，然后算答案即可

Code

#include<cstdio>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define ll long long
using namespace std;
int a[201000],n;
int main()
{
    scanf("%d",&n);fo(i,1,n) scanf("%d",&a[i]),a[i+n]=a[i];int i;
    for(i=1;i<=n;i++){
        bool bz=1;int k=0;
        fo(j,i,i+n-1) {k+=a[j];if(k<j-i+1) {bz=0;break;}}
        if(bz) break;
    }
    int j=i,l=i;ll ans=0;
    fo(k,i,i+n-1){
        if(a[k]>0) for(;j<=l+a[k]-1;j++) ans+=(ll)(j-k)*(j-k);
        l=j;
    }
    printf("%lld",ans);
}