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

Description

ZY带N个小Kid过河,小KID分成两种:高一年级,高二年级,由于存在代沟问题,如果同一条船上高一年级生和高二年级生数量之差超过K,就会发生不和谐的事件.当然如果一条船上全是同一年级的,就绝对不会发生争执.现在ZY按小KID队列的顺序依次安排上船,并且不能让他们在过河时发生争执.对于当前等待上船的小KID来说,要么让他上船,要么将停在渡口的船开走,再让他上另一条船,同一条船上的人不过超过M人.为了让所有的小KID过河,在知悉小KID队列的情况下,最少需要多少条船.

Input

第一行给出N,M,K.含义如上所述 下行N行用来描述小KID的队列,每行一个字符”A”或者”L”

Output

最少需要多少条船

Sample Input

5 4 1
A
L
L
L
A

Sample Output

2

HINT

前三个人一条船,后两个人一条船
数据范围
30% 数据中1<=N<=1000
100%数据中1<=N<=250000,1<=M,K<=N

———————————————————————

这道题我们用单调队列维护相同一段的情况 用平衡树维护高一高二差不能超过k的情况

相同一段不能超过m的明显符合所选区间是在不断挪动的 所以用单调队列

而相差的话我们可以转换为类似扫描线的东西

先维护一波前缀和（sum） 是A就+1 L -1 这样方便查询差

转移方程 f【x】=miin （f【k】+1）

x-m<=k<=x  sum【x】-m<=sum【k】<=sum【x】+m

这样我们可以按坐标x扫 查询区间 max

但是这里的线段树我们下标是 sum

同一个下标可能有多个点 所以对于线段树的每个叶子节点我们都要建一棵平衡树

所以我们的操作是 插入 删除 区间取max （平衡树/线段树）

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