码迷,mamicode.com
首页 > 编程语言 > 详细

ZJOI2017 树状数组

时间:2019-01-04 17:57:50      阅读:168      评论:0      收藏:0      [点我收藏+]

标签:zjoi   现在   思路   问题   names   static   --   false   code   

属于可怜出的小清新数据结构题呢
题目链接


解析

因为全部都在模\(2\)意义下,因此相当于单点异或,查询区间异或和.
如果你对树状数组足够熟悉,那么你会发现可怜写了一个单点加求后缀和的程序.
因此\([l,r]\)正确的概率就要使\(a_{l-1}\oplus a_l\oplus a_{l+1}\oplus ...\oplus a_n=a_r\oplus a_{r+1}\oplus...\oplus a_n\)
\(a_{l-1}=a_r\)
于是我们用线段树维护这个问题好像就做完了
当然不对.
假如我们在\([2,3]\)中等概率取反,然后查询\(a_2\)\(a_3\)相等的概率,江道理应该是\(0\),但是我们的线段树会得到\(\frac{1}{2}\).

因此我们用一个二维点\(f(x,y)\)表示\(a_x\)\(a_y\)相等的概率.

修改

假设我们操作\([l,r]\),那么它会对多少点造成影响呢?
假如有一个点\((x,y)\),满足\(l\leq x\leq y\leq r\),那么会有\(\frac{2}{r-l+1}\)的概率使\((x,y)\)的相同性取反(即相同变不同,不同变相同).
那么操作过一次之后\(f(x,y)=f(x,y)*(1-\frac{2}{r-l+1})+(1-f(x,y))*\frac{2}{r-l+1}\).
这个式子还是比较显然的.

假如有一个点\((x,y)\),满足\(x\leq l\leq y\leq r\),那么会有\(\frac{1}{r-l+1}\)的概率使\((x,y)\)的相同性取反(即相同变不同,不同变相同).
那么操作过一次之后\(f(x,y)=f(x,y)*(1-\frac{1}{r-l+1})+(1-f(x,y))*\frac{1}{r-l+1}\).
这个式子也是比较显然的.

对于\(l\leq x\leq r\leq y\)同理.

询问

直接输出\(f(x,y)\)即可.

实现

现在思路已经比较明显了.
对于每次修改,我们相当于修改\(3\)个矩形.
对于每次询问,我们相当于询问一个点的值.
树套树实现即可.
因为是单点询问,标记永久化.

坑点

注意到当查询为\(0\)时,会直接返回\(0\).这种情况相当于前缀异或等于后缀异或的概率,注意特判.

代码如下

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#define N (100010)
#define P (998244353)
#define inf (0x7f7f7f7f)
#define rg register int
#define Label puts("NAIVE")
#define spa print(' ')
#define ent print('\n')
#define rand() (((rand())<<(15))^(rand()))
typedef long double ld;
typedef long long LL;
typedef unsigned long long ull;
using namespace std;
inline char read(){
    static const int IN_LEN=1000000;
    static char buf[IN_LEN],*s,*t;
    return (s==t?t=(s=buf)+fread(buf,1,IN_LEN,stdin),(s==t?-1:*s++):*s++);
}
template<class T>
inline void read(T &x){
    static bool iosig;
    static char c;
    for(iosig=false,c=read();!isdigit(c);c=read()){
        if(c=='-')iosig=true;
        if(c==-1)return;
    }
    for(x=0;isdigit(c);c=read())x=((x+(x<<2))<<1)+(c^'0');
    if(iosig)x=-x;
}
inline char readchar(){
    static char c;
    for(c=read();!isalpha(c)&&!isdigit(c);c=read())
    if(c==-1)return 0;
    return c;
}
const int OUT_LEN = 10000000;
char obuf[OUT_LEN],*ooh=obuf;
inline void print(char c) {
    if(ooh==obuf+OUT_LEN)fwrite(obuf,1,OUT_LEN,stdout),ooh=obuf;
    *ooh++=c;
}
template<class T>
inline void print(T x){
    static int buf[30],cnt;
    if(x==0)print('0');
    else{
        if(x<0)print('-'),x=-x;
        for(cnt=0;x;x/=10)buf[++cnt]=x%10+48;
        while(cnt)print((char)buf[cnt--]);
    }
}
inline void flush(){fwrite(obuf,1,ooh-obuf,stdout);}
struct xds{
    int ls,rs,w;
}a[35000010];
int rt,n,m,ind,ind2,ans,inv[N],T[N<<2];
int ksm(int a,int p){
    int res=1;
    while(p){
        if(p&1)res=1ll*res*a%P;
        a=1ll*a*a%P,p>>=1;
    }
    return res;
}
void merge(int &x,int y){x=((2ll*x*y%P-x-y+1)%P+P)%P;}
void Modify(int &x,int L,int R,int l,int r,int w){
    if(!x)x=++ind2,a[x].w=1;
    if(L==l&&R==r)return merge(a[x].w,w);
    int mid=(L+R)>>1;
    if(r<=mid)Modify(a[x].ls,L,mid,l,r,w);
    else if(l>mid)Modify(a[x].rs,mid+1,R,l,r,w);
    else Modify(a[x].ls,L,mid,l,mid,w),Modify(a[x].rs,mid+1,R,mid+1,r,w);
}
void modify(int &x,int L,int R,int lx,int ly,int rx,int ry,int w){
    if(!x)x=++ind;
    if(L==lx&&R==ly)return Modify(T[x],1,n,rx,ry,w);
    int mid=(L+R)>>1;
    if(ly<=mid)modify(a[x].ls,L,mid,lx,ly,rx,ry,w);
    else if(lx>mid)modify(a[x].rs,mid+1,R,lx,ly,rx,ry,w);
    else modify(a[x].ls,L,mid,lx,mid,rx,ry,w),modify(a[x].rs,mid+1,R,mid+1,ly,rx,ry,w); 
}
void Query(int x,int L,int R,int p){
    if(!x)return;merge(ans,a[x].w);
    if(L==R)return;int mid=(L+R)>>1;
    if(p<=mid)Query(a[x].ls,L,mid,p);
    else Query(a[x].rs,mid+1,R,p);
}
void query(int x,int L,int R,int px,int py){
    if(!x)return;if(T[x])Query(T[x],1,n,py); 
    if(L==R)return;int mid=(L+R)>>1;
    if(px<=mid)query(a[x].ls,L,mid,px,py);
    else query(a[x].rs,mid+1,R,px,py);
}
int main(){
    read(n),read(m),ind2=400000; 
    for(int i=0;i<=n;i++)inv[i]=ksm(i,P-2);
    for(int t=0,op,l,r;m;m--){
        read(op),read(l),read(r);
        if(op==1){
            int len=r-l+1; t++;
            modify(rt,0,n,l,r,l,r,(1-2*inv[len]%P+P)%P);
            modify(rt,0,n,0,l-1,l,r,(1-inv[len]+P)%P);
            if(r+1<=n)modify(rt,0,n,l,r,r+1,n,(1-inv[len]+P)%P);
        }
        else{
            ans=1,query(rt,0,n,l-1,r);
            if(l==1&&(t&1))ans=(1-ans+P)%P;
            print(ans),ent;
        }
    }
    return flush(),0;
}

ZJOI2017 树状数组

标签:zjoi   现在   思路   问题   names   static   --   false   code   

原文地址:https://www.cnblogs.com/Romeolong/p/10221221.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!