bzoj4785: [Zjoi2017]树状数组

传送门
老司机应该能发现他求的是后缀和吧。
然后就变成了判断a[l-1]==a[r]的概率。
区间覆盖的二维线段树水过。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#define N 100005
#define ll long long
#define mo 998244353
#define merge(a,b) ((a*b+(1-a+mo)*(1-b+mo))%mo)
using namespace std;
int ls[330*N],rs[330*N],rt[4*N],sz;
ll sum[330*N],ans;
int n,m,tp,l,r;
inline ll power(ll a,ll b){
    ll s=1;
    for (;b;b/=2,a=a*a%mo)
        if (b&1) s=s*a%mo;
    return s;
}
inline void change(int &x,int l,int r,int l1,int r1,ll v){
    if (!x) sum[x=++sz]=1;
    if (l1<=l&&r<=r1){
        sum[x]=merge(sum[x],v);
        return;
    }
    int mid=(l+r)/2;
    if (r1<=mid) change(ls[x],l,mid,l1,r1,v);
    else if (l1>mid) change(rs[x],mid+1,r,l1,r1,v);
    else change(ls[x],l,mid,l1,mid,v),change(rs[x],mid+1,r,mid+1,r1,v);
}
inline void ask(int x,int l,int r,int p){
    if (!x) return;
    ans=merge(ans,sum[x]);
    if (l==r) return;
    int mid=(l+r)/2;
    if (p<=mid) ask(ls[x],l,mid,p);
    else ask(rs[x],mid+1,r,p);
}
inline void change(int x,int l,int r,int l1,int r1,int l2,int r2,ll v){
    if (l1<=l&&r<=r1){
        change(rt[x],0,n+1,l2,r2,v);
        return;
    }
    int mid=(l+r)/2;
    if (r1<=mid) change(x*2,l,mid,l1,r1,l2,r2,v);
    else if (l1>mid) change(x*2+1,mid+1,r,l1,r1,l2,r2,v);
    else change(x*2,l,mid,l1,mid,l2,r2,v),
        change(x*2+1,mid+1,r,mid+1,r1,l2,r2,v);
}
inline void ask(int x,int l,int r,int p,int q){
    if (rt[x]) ask(rt[x],0,n+1,q);
    if (l==r) return;
    int mid=(l+r)/2;
    if (p<=mid) ask(x*2,l,mid,p,q);
    else ask(x*2+1,mid+1,r,p,q);
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++){
        scanf("%d%d%d",&tp,&l,&r);
        if (tp==1){
            ll p=power(r-l+1,mo-2);
            if (l>1) change(1,0,n,1,l-1,l,r,(1-p+mo)%mo);
            if (r<n) change(1,0,n,l,r,r+1,n,(1-p+mo)%mo);
            ll pp=(p*2)%mo;
            change(1,0,n,l,r,l,r,(1-pp+mo)%mo);
            change(1,0,n,0,0,0,l-1,0);
            change(1,0,n,0,0,r+1,n+1,0);
            change(1,0,n,0,0,l,r,p);
        }
        else{
            ans=1;
            ask(1,0,n,l-1,r);
            printf("%lld\n",ans);
        }
    }
}