1. 程式人生 > >【ZJOI2017】樹狀數組

【ZJOI2017】樹狀數組

\n tag amp class () 不同的 pre clu cst

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)) << \n; else { std::cout << (sum?d2::query(rt2,r):reduce(1-d2::query(rt2,r))) << \n; } } } }

【ZJOI2017】樹狀數組