ZJOI2017树状数组

Posted 2021-02-06 skip1978

sts维护两个数不同的概率就可以了

#include<cstdio>
#include<iostream>
#include<algorithm>
const int mod = 998244353;
typedef long long ll;
inline int pow(int x,int y,int ans=1){
    for(;y;y>>=1,x=ll(x)*x%mod)
        if(y&1)ans=ll(ans)*x%mod;
    return ans;
}
inline int inv(int x){return pow(x,mod-2);}
int n,m;
inline int reduce(int x){return x + (x>>31&mod);}
inline int merge(int x,int y){return (ll(mod+1-x)*y+ll(mod+1-y)*x)%mod;}
const int maxn = 100100;
namespace d2{
    const int maxm = maxn * 400;
    int L[maxm],R[maxm],tag[maxm],tot;
    inline void modify(int&rt,int ll,int rr,int v,int l=1,int r=n){
        if(!rt)rt=++tot;
        if(ll<=l&&r<=rr)return void(tag[rt]=merge(tag[rt],v));
        int mid=l+r>>1;
        if(ll<=mid)modify(L[rt],ll,rr,v,l,mid);
        if(mid<rr)modify(R[rt],ll,rr,v,mid+1,r);
    }
    inline int query(int&rt,int pos,int l=1,int r=n){
        if(rt==0||l==r)return tag[rt];
        int mid=l+r>>1,ans;
        if(pos<=mid)ans=query(L[rt],pos,l,mid);
        else ans=query(R[rt],pos,mid+1,r);
        return merge(ans,tag[rt]);
    }
}
const int maxm = maxn * 20;
int L[maxm],R[maxm],val[maxm],tot;
inline void add(int&rt,int ll,int rr,int ll2,int rr2,int v,int l=1,int r=n){
    if(!rt)rt=++tot;
    if(ll<=l&&r<=rr)return d2::modify(val[rt],ll2,rr2,v);
    int mid=l+r>>1;
    if(ll<=mid)add(L[rt],ll,rr,ll2,rr2,v,l,mid);
    if(mid<rr)add(R[rt],ll,rr,ll2,rr2,v,mid+1,r);
}
inline int query(int&rt,int pos1,int pos2,int l=1,int r=n){
    if(rt==0||l==r)return d2::query(val[rt],pos2);
    int mid=l+r>>1,ans;
    if(pos1<=mid)ans=query(L[rt],pos1,pos2,l,mid);
    else ans=query(R[rt],pos1,pos2,mid+1,r);
    return merge(ans,d2::query(val[rt],pos2));
}
int rt,rt2;
int main(){
    std::ios::sync_with_stdio(false),std::cin.tie(0);
    std::cin >> n >> m;
    for(int i=1,l,r,opt,sum=0;i<=m;++i){
        std::cin >> opt >> l >> r;
        if(opt == 1){
            const int p1 = inv(r-l+1),p2=p1*2%mod;
            add(rt,l,r,l,r,p2);
            if(r!=n)add(rt,l,r,r+1,n,p1);
            if(l!=1)add(rt,1,l-1,l,r,p1);
            d2::modify(rt2,l,r,p1);
            sum^=1;
        }else{
            if(l!=1)
                std::cout << reduce(1-query(rt,l-1,r)) << ‘
‘;
            else {
                std::cout << (sum?d2::query(rt2,r):reduce(1-d2::query(rt2,r))) << ‘
‘;
            }
        }
    }
}