bzoj 1314: River过河 树套树+单调队列

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
using namespace std;
const int M=250007,N=1<<19,inf=0x3f3f3f3f;
int read(){
    int ans=0,f=1,c=getchar();
    while(c<‘0‘||c>‘9‘){if(c==‘-‘) f=-1; c=getchar();}
    while(c>=‘0‘&&c<=‘9‘){ans=ans*10+(c-‘0‘); c=getchar();}
    return ans*f;
}
multiset<int> q[N];
int x,n,m,k,sum[M];
int qq[N],ql,qr=1;
char str[M];
int s[N<<1],f[M];
void modify(int x){
    int v=q[x].size()?*q[x].begin():inf;
    for(s[x+=N]=v,x>>=1;x;x>>=1) s[x]=min(s[x<<1],s[x<<1^1]);
}
int query(int l,int r){
    int mn=inf;
    for(l+=N-1,r+=N+1;r-l!=1;l>>=1,r>>=1){
        if(~l&1) mn=min(mn,s[l+1]);
        if(r&1)  mn=min(mn,s[r-1]);
    }
    return mn;
}
int main(){
    memset(s,0x3f,sizeof(s));
    n=read(); m=read(); k=read();
    sum[0]=N/2;
    for(int i=1;i<=n;i++){
        scanf("%s",str+i);
        sum[i]=sum[i-1]+(str[i]==‘A‘?1:-1);
    }
    q[sum[0]].insert(0);
    modify(sum[0]);
    for(int i=1,v,j=1;i<=n;i++){
        while(ql<=qr&&(qq[ql]<i-m||str[qq[ql]+1]!=str[i])) ++ql;
        while(ql<=qr&&f[qq[qr]]>f[i-1]) --qr;
        qq[++qr]=i-1;
        v=query(max(1,sum[i]-k),min(N-2,sum[i]+k));
        f[i]=min(v,ql<=qr?f[qq[ql]]:inf)+1;
        if(i-m>=0){
            v=sum[i-m];
            q[v].erase(q[v].find(f[i-m]));
            modify(v);
        }
        v=sum[i];
        q[v].insert(f[i]);
        modify(v);
    }
    printf("%d\n",f[n]);
    return 0;
}