BZOJ 4412/Usaco2016 Feb Circular Barn(構造)

思路
一下是來自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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章