BZOJ 1563 诗人小G（四边形优化）

struct node
{
    int L,R,id;


    node(){}
    node(int _L,int _R,int _id)
    {
        L=_L;
        R=_R;
        id=_id;
    }
};


node Q[N];
int s[N],n,L,P;
long double F[N];


long double C(int i,int j)
{
    return F[i]+pow(fabs(s[j]-s[i]+j-i-1-L),P);
}


void DP()
{
    int H=0,T=0;
    Q[0]=node(1,n,0);
    int i,L,R,mid,ans;
    FOR1(i,n)
    {
        while(i>Q[H].R) H++;
        F[i]=C(Q[H].id,i);
        if(C(i,n)>C(Q[T].id,n)) continue;
        while(i<Q[T].L&&C(i,Q[T].L)<C(Q[T].id,Q[T].L)) T--;
        
        L=max(Q[T].L,i+1);
        R=Q[T].R;
        ans=min(n,Q[T].R+1);
        while(L<=R)
        {
            mid=(L+R)>>1;
            if(C(i,mid)<C(Q[T].id,mid)) R=mid-1,ans=mid;
            else L=mid+1;
        }
        Q[T].R=ans-1;
        Q[++T]=node(ans,n,i);
    }
    if(F[n]>dinf)
    {
        puts("Too hard to arrange");
    }
    else printf("%lld\n",(i64)F[n]);
    puts("--------------------");
}

int get()
{
    int L=0;
    char c=getchar();
    while(c!=‘\n‘)
    {
        L++;
        c=getchar();
    }
    return L;
}


int main()
{
    rush()
    {
        RD(n,L,P);
        char c=getchar();
        while(c!=‘\n‘) c=getchar();
        int i;
        FOR1(i,n) s[i]=get()+s[i-1];
        DP();
    }
}