• [ZJOI2017]树状数组


    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,因此第一次询问可怜一定正确,第二次询问可怜一定错误。

    数据范围

     查询(l,r)从前缀变为了后缀

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

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

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

    当l=1时

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

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

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

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

    1.使询问的值一样的概率:

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

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

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

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

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

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

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

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

    不过空间不够,要动态开点

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

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

    都不相同合起来就相同了

     1 #include<iostream>
     2 #include<cstdio>
     3 #include<cstring>
     4 #include<algorithm>
     5 #include<cmath>
     6 using namespace std;
     7 typedef long long lol;
     8 lol Mod=998244353;
     9 int size,root[400001],ch[30000001][2],n,m;
    10 int ans,sum[30000001];
    11 int merge(lol x,lol y)
    12 {
    13   return (x*y%Mod+(1-x+Mod)*(1-y+Mod)%Mod)%Mod;
    14 }
    15 lol qpow(lol x,int y)
    16 {
    17   lol res=1;
    18   while (y)
    19     {
    20       if (y&1) res=res*x%Mod;
    21       x=x*x%Mod;
    22       y>>=1;
    23     }
    24   return res;
    25 }
    26 void update2(int &rt,int l,int r,int L,int R,lol v)
    27 {
    28   if (!rt) rt=++size,sum[rt]=1;
    29   if (l>=L&&r<=R)
    30     {
    31       sum[rt]=merge(sum[rt],v);
    32       return;
    33     }
    34   int mid=(l+r)>>1;
    35   if (L<=mid) update2(ch[rt][0],l,mid,L,R,v);
    36   if (R>mid) update2(ch[rt][1],mid+1,r,L,R,v);
    37 }
    38 void update1(int rt,int l,int r,int L,int R,int LL,int RR,lol v)
    39 {
    40   if (l>=L&&r<=R)
    41     {
    42       update2(root[rt],0,n+1,LL,RR,v);
    43       return;
    44     }
    45   int mid=(l+r)>>1;
    46   if (L<=mid) update1(rt<<1,l,mid,L,R,LL,RR,v);
    47   if (R>mid) update1(rt<<1|1,mid+1,r,L,R,LL,RR,v);
    48 }
    49 void query2(int rt,int l,int r,int x1)
    50 {
    51   if (!rt) return;
    52   ans=merge(ans,sum[rt]);
    53   if (l==r) return;
    54   int mid=(l+r)>>1;
    55   if (x1<=mid) query2(ch[rt][0],l,mid,x1);
    56   else query2(ch[rt][1],mid+1,r,x1);
    57 }
    58 void query1(int rt,int l,int r,int x1,int x2)
    59 {
    60   if (root[rt]) query2(root[rt],0,n+1,x2);
    61   if (l==r)
    62       return;
    63   int mid=(l+r)>>1;
    64   if (x1<=mid) query1(rt<<1,l,mid,x1,x2);
    65   else query1(rt<<1|1,mid+1,r,x1,x2);
    66 }
    67 int main()
    68 {int i,opt,l,r;
    69   cin>>n>>m;
    70   for (i=1;i<=m;i++)
    71     {
    72       scanf("%d%d%d",&opt,&l,&r);
    73       if (opt==1)
    74     {
    75       lol p=qpow(r-l+1,Mod-2);
    76       if (l>1) update1(1,0,n,1,l-1,l,r,(1-p+Mod)%Mod);
    77       if (r<n) update1(1,0,n,l,r,r+1,n,(1-p+Mod)%Mod);
    78       lol pp=p*2%Mod;
    79       update1(1,0,n,l,r,l,r,(1-pp+Mod)%Mod);
    80       update1(1,0,n,0,0,0,l-1,0);
    81       update1(1,0,n,0,0,r+1,n+1,0);
    82       update1(1,0,n,0,0,l,r,p);
    83     }
    84       else
    85     {
    86       ans=1;
    87       query1(1,0,n,l-1,r);
    88       printf("%d
    ",ans);
    89     }
    90     }
    91 }
  • 相关阅读:
    堆排序
    分治与递归的结合-------快速排序
    递归算法--写实例----阶乘问题---整数划分问题
    PAT A+B格式
    学习C#的第一天
    C#快捷键
    字符串匹配算法(在字符串T中查找是否有与字符串P相同的子串)
    linux中nohup实时查看启动日志的解决办法
    linux 产看pycharm 安装路径
    pip requirements.txt
  • 原文地址:https://www.cnblogs.com/Y-E-T-I/p/8317487.html
Copyright © 2020-2023  润新知