P5445 [APIO2019]路灯（树套树）

P5445 [APIO2019]路灯

转化为平面上的坐标(x,y)，set维护连续区间.

用树套树维护矩阵加法，单点查询。

注意维护矩阵差分的时候，

$(x,y,v)$是对$(x,y)(n+1,n+1)$的矩阵做出贡献

#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
#define ri register int
using namespace std;
int read(){
    char c=getchar(); ri x=;
    while(c<''||c>'') c=getchar();
    while(''<=c&&c<='') x=x*+c-,c=getchar();
    return x;
}
#define N 600005
#define W 20000005
struct E{
    int l,r;
    E(int A,int B):l(A),r(B){}
    bool operator < (const E &G) const{return r<G.r;}
};set<E> g;
set<E>::iterator it;
int n,T,a[N]; char opt[],cc[N];
int cnt,rt[N],lc[W],rc[W],s[W];
#define mid (l+r)/2
void Ins1(int &o,int l,int r,int x,int v){
    if(!o)o=++cnt; s[o]+=v;
    if(l==r) return;
    if(x<=mid) Ins1(lc[o],l,mid,x,v);
    else Ins1(rc[o],mid+,r,x,v);
}
int Ask1(int o,int l,int r,int x1,int x2){
    if(!o||x2<l||r<x1) return ;
    if(x1<=l&&r<=x2) return s[o];
    return Ask1(lc[o],l,mid,x1,x2)+Ask1(rc[o],mid+,r,x1,x2);
}
void Ins2(int x,int y,int v){for(;x<=n+;x+=x&-x)Ins1(rt[x],,n+,y,v);}
int Ask2(int x,int y){int re=; for(;x;x-=x&-x)re+=Ask1(rt[x],,n+,,y); return re;}
inline void Add(int x1,int y1,int x2,int y2,int v){//注意修改标记是对右上的查询产生贡献，并不是平移区间
    Ins2(x1,y1,v),Ins2(x1,y2+,-v),Ins2(x2+,y1,-v),Ins2(x2+,y2+,v);
}
int main(){
    n=read()+; T=read(); scanf("%s",cc+);
    for(ri i=;i<=n;++i) g.insert(E(i,i));
    int x,y,pl,pr,ql,qr;
    for(ri i=;i<n;++i) if((a[i]=cc[i]-)){
        it=g.lower_bound(E(,i+)); --it; pl=(*it).l;
        g.erase(it); g.erase(E(i+,i+)); g.insert(E(pl,i+));
    }
    for(it=g.begin();it!=g.end();++it) Add((*it).l,(*it).l,(*it).r,(*it).r,T);
    for(ri t=;t<=T;++t){
        scanf("%s",opt);
        if(opt[]=='q'){
            x=read(); y=read();
            pl=(*(g.lower_bound(E(,x)))).l;
            ql=(*(g.lower_bound(E(,y)))).l;
            printf("%d\n",Ask2(x,y)-(T-t)*(pl==ql));
        }else{
            x=read();
            if(a[x]){
                it=g.lower_bound(E(,x));
                pl=(*it).l,pr=x; ql=x+,qr=(*it).r;
                Add(pl,ql,pr,qr,t-T);
                g.erase(it); g.insert(E(pl,pr)); g.insert(E(ql,qr));
            }else{
                it=g.lower_bound(E(,x));
                pl=(*it).l,pr=x; ++it; ql=x+,qr=(*it).r;
                Add(pl,ql,pr,qr,T-t);
                g.erase(E(pl,pr)); g.erase(E(ql,qr)); g.insert(E(pl,qr));
            }a[x]^=;
        }
    }return ;
}