您的位置:首页 > 其它

[ZJOI2017]树状数组

2018-01-19 16:10 429 查看

Description

漆黑的晚上,九条可怜躺在床上辗转反侧。难以入眠的她想起了若干年前她的一次悲惨的OI 比赛经历。那是一道

基础的树状数组题。给出一个长度为 n 的数组 A,初始值都为 0,接下来进行 m 次操作,操作有两种:
1 x,表示将 Ax 变成 (Ax + 1) mod 2。
2 l r,表示询问 sigma(Ai) mod 2,L<=i<=r
尽管那个时候的可怜非常的 simple,但是她还是发现这题可以用树状数组做。当时非常young 的她写了如下的算
法:



其中 lowbit(x) 表示数字 x 最?的非 0 二进制位,例如 lowbit(5) = 1, lowbit(12) = 4。进行第一类操作的时
候就调用 Add(x),第二类操作的时候答案就是 Query(l, r)。如果你对树状数组比较熟悉,不难发现可怜把树状
数组写错了: Add和Find 中 x 变化的方向反了。因此这个程序在最终测试时华丽的爆 0 了。然而奇怪的是,在
当时,这个程序通过了出题人给出的大样例——这也是可怜没有进行对拍的原因。现在,可怜想要算一下,这个程
序回答对每一个询问的概率是多少,这样她就可以再次的感受到自己是一个多么非的人了。然而时间已经过去了很
多年,即使是可怜也没有办法完全回忆起当时的大样例。幸运的是,她回忆起了大部分内容,唯一遗忘的是每一次
第一类操作的 x的值,因此她假定这次操作的 x 是在 [li, ri] 范围内 等概率随机 的。具体来说,可怜给出了
一个长度为 n 的数组 A,初始为 0,接下来进行了 m 次操作:
1 l r,表示在区间 [l, r] 中等概率选取一个 x 并执行 Add(x)。
2 l r,表示询问执行 Query(l, r) 得到的结果是正确的概率是多少。

Input

第一行输入两个整数 n, m。
接下来 m 行每行描述一个操作,格式如题目中所示。
N<=10^5,m<=10^5,1<=L<=R<=N

Output

对于每组询问,输出一个整数表示答案。如果答案化为最简分数后形如 x/y

,那么你只需要输出 x*y-1 mod 998244353 后的值。(即输出答案模 998244353)。

Sample Input

5 5

1 3 3

2 3 5

2 4 5

1 1 3

2 2 5

Sample Output

1

0

665496236

//在进行完 Add(3) 之后, A 数组变成了 [0, 1, 1, 0, 0]。所以前两次询问可怜的程序答案都是

1,因此第一次询问可怜一定正确,第二次询问可怜一定错误。

数据范围



[b]查询(l,r)从前缀变为了后缀[/b]

[b]那么原来是S[r]-s[l-1],变成了S[l-1]-s[r][/b]

[b]原来的区间是[l,r],现在变成了[l-1,r-1][/b]

[b]询问要正确就要求l-1和r的值必须一样[/b]

[b]当l=1时[/b]

[b]S[r]-s[0]=>s[0]-s[r]=-s[r][/b]

[b]所以此时要正确就必须使1~r的前缀和与r~n的前缀和相同[/b]

[b]将查询(l-1,r)用一个点表示[/b]

[b]修改[l,r]的一个数,分成几种情况讨论:[/b]

[b]1.使询问的值一样的概率:[/b]

[b](1):x属于[1,l-1],y属于[l,r].此时要相同必须要求y不被选中,概率(1-p)[/b]

[b](2):y属于[r+1,n],x属于[l,r].此时同上,x不能被选中[/b]

[b](3):x,y同属于[l,r].x,y都不能被选中,概率(1-2p)[/b]

[b]2.使询问的r的前缀和等于后缀和的概率:x为0[/b]

[b](1):y属于[0,l-1].因为[l,r]总会改一个数,所以概率为0[/b]

[b](2):y属于[r+1,n+1].同上[/b]

[b](3):y属于[l,r].要求选中y,概率为p[/b]

[b]修改用树套树(二维线段树),外层的树维护第一维坐标,内层的树维护第二维坐标[/b]

[b]不过空间不够,要动态开点[/b]

[b]如何合并两次修改:假设两次的相同概率分别为p1,p2(以上2种情况统称相同)[/b]

[b]显然新的概率为:p1*p2+(1-p1)*(1-p2)[/b]

[b]都不相同合起来就相同了[/b]

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long lol;
lol Mod=998244353;
int size,root[400001],ch[30000001][2],n,m;
int ans,sum[30000001];
int merge(lol x,lol y)
{
return (x*y%Mod+(1-x+Mod)*(1-y+Mod)%Mod)%Mod;
}
lol qpow(lol x,int y)
{
lol res=1;
while (y)
{
if (y&1) res=res*x%Mod;
x=x*x%Mod;
y>>=1;
}
return res;
}
void update2(int &rt,int l,int r,int L,int R,lol v)
{
if (!rt) rt=++size,sum[rt]=1;
if (l>=L&&r<=R)
{
sum[rt]=merge(sum[rt],v);
return;
}
int mid=(l+r)>>1;
if (L<=mid) update2(ch[rt][0],l,mid,L,R,v);
if (R>mid) update2(ch[rt][1],mid+1,r,L,R,v);
}
void update1(int rt,int l,int r,int L,int R,int LL,int RR,lol v)
{
if (l>=L&&r<=R)
{
update2(root[rt],0,n+1,LL,RR,v);
return;
}
int mid=(l+r)>>1;
if (L<=mid) update1(rt<<1,l,mid,L,R,LL,RR,v);
if (R>mid) update1(rt<<1|1,mid+1,r,L,R,LL,RR,v);
}
void query2(int rt,int l,int r,int x1)
{
if (!rt) return;
ans=merge(ans,sum[rt]);
if (l==r) return;
int mid=(l+r)>>1;
if (x1<=mid) query2(ch[rt][0],l,mid,x1);
else query2(ch[rt][1],mid+1,r,x1);
}
void query1(int rt,int l,int r,int x1,int x2)
{
if (root[rt]) query2(root[rt],0,n+1,x2);
if (l==r)
return;
int mid=(l+r)>>1;
if (x1<=mid) query1(rt<<1,l,mid,x1,x2);
else query1(rt<<1|1,mid+1,r,x1,x2);
}
int main()
{int i,opt,l,r;
cin>>n>>m;
for (i=1;i<=m;i++)
{
scanf("%d%d%d",&opt,&l,&r);
if (opt==1)
{
lol p=qpow(r-l+1,Mod-2);
if (l>1) update1(1,0,n,1,l-1,l,r,(1-p+Mod)%Mod);
if (r<n) update1(1,0,n,l,r,r+1,n,(1-p+Mod)%Mod);
lol pp=p*2%Mod;
update1(1,0,n,l,r,l,r,(1-pp+Mod)%Mod);
update1(1,0,n,0,0,0,l-1,0);
update1(1,0,n,0,0,r+1,n+1,0);
update1(1,0,n,0,0,l,r,p);
}
else
{
ans=1;
query1(1,0,n,l-1,r);
printf("%d\n",ans);
}
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: