思路
一下是來自lxe神犇的寫法:
首先我們先破環成鏈。。。
將所有的值減掉1剩下的就表示多出來的
由於每次都是往順時針方向走(一開始看錯題了)因此我們找到一個能夠擴展到最遠的起點
即不會有其他的牛走過這裏 證明yy一下 暫時寫不出來
於是我們先求一個最大的連續子段和 找到起點 然後暴力往下推就行了
順便一提的是 (x+y)^2>x^2+y^2;
所以如果一個位置有牛 且新來一隻 需要替換一下接着走;
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
#include<vector>
using namespace std;
typedef long long LL;
const int imax=1000000+229;
int n,a[imax*2]; LL sum[imax];
void iread()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]),a[n+i]=a[i];
}
int lishi[imax*2],x[imax*2];
int inext(int x){ ++x; if(x<=2*n) return x; else return 1;}
void iwork()
{
int temp=0; int begin=0; int last=0; int Max=0;
for(int i=1;i<=n*2;i++) lishi[i]=a[i]-1;
for(int i=1;i<=n*2;i++)
{
temp+=lishi[i];
if(temp>Max) { Max=temp; begin=last;}
if(temp<0) { temp=0; last=i+1; }
}
// printf("%d\n",begin);
int tot=1;
if(begin>n) begin-=n;
for(int i=begin;tot<=n;i=inext(i)) x[tot++]=lishi[i];
tot--;
for(int i=1;i<=n;i++) sum[i]=sum[i-1]+(LL)i*i;
// for(int i=1;i<=tot;i++) printf("%d\n",x[i]);
int k=1; LL ans=0;
ans+=sum[x[1]]; k+=x[1];
for(int i=2;i<=n;i++)
{
if(k>=i) x[i]++;
if(x[i]>0)
{
if(k<i) k=i;
int pos=k+x[i];
ans+=(sum[pos-i]-sum[k-i]);
k=pos;
}
}
printf("%lld\n",ans);
}
int main()
{
// freopen("p3.in","r",stdin);
iread();
iwork();
return 0;
}