ZOJ 3911 线段树

题意：有N个数字，M个操作，然后回答每个Q开头的询问

操作形式：

A val pos：在pos位置上+val

Q l r：询问l~r之间有多少个质数

R val l r：把l~r之间的数字替换成val

分析：建成两棵树，一棵更新 原数列，一棵更新 质数序列（只有0,1）

 #include <cstdio>
 #include <cstring>
 #include <cmath>
 #include <algorithm>
 using namespace std;
 #define repu(i,a,b) for(int i=a;i<b;i++)
 #define lson l , m , rt << 1
 #define rson m + 1 , r , rt << 1 | 1
 const int maxn = ;
 const int M = ;
 int h , w , n;
 bool p[M];
 int prime()
 {
     memset(p,false,sizeof(p));
     p[] = p[] = true;
     for(int i = ; i <= M; i += )
         p[i] = true;
     for(int i = ; i <= sqrt(M); i+=)
     {
         int k = i * ;
         for(int j = i*i; j <= M; j += k)
             p[j] = true;
     }
 }
 struct Segment
 {
     int col[maxn<<];
     int sum[maxn<<];
     void init()
     {
         memset(col,,sizeof(col));
         memset(sum,,sizeof(sum));
     }
     void PushUp(int rt)
     {
         sum[rt] = sum[rt<<] + sum[rt<<|];
     }
     void PushDown(int rt,int m)
     {
         if(col[rt] != -)///如果是0就进不去
         {
             col[rt<<] = col[rt<<|] = col[rt];
             sum[rt<<] = (m - (m >> )) * col[rt];
             sum[rt<<|] = (m >> ) * col[rt];
             col[rt] = -;
         }
     }
     void build(int l,int r,int rt)
     {
         col[rt] = -;
         sum[rt] = ;
         if (l == r) return ;
         int m = (l + r) >> ;
         build(lson);
         build(rson);
         PushUp(rt);
     }
     void Update(int p,int add,int l,int r,int rt)
     {
         if(l == r)
         {
             sum[rt] += add;
             return ;
         }
         PushDown(rt , r - l + );
         int m = (l + r) >> ;
         if (p <= m) Update(p, add, lson);
         else Update(p, add, rson);
         PushUp(rt);
     }
     void update(int L,int R,int c,int l,int r,int rt)
     {
         if (L <= l && r <= R)
         {
             col[rt] = c;
             ///因为序列只有0,1;如果初始化为0的话，赋值有可能赋成0，pushdown就进不了，也就跟新不了数据了
             sum[rt] = c * (r - l + );
             return ;
         }
         PushDown(rt , r - l + );
         int m = (l + r) >> ;
         if (L <= m) update(L , R , c , lson);
         if (R > m) update(L , R , c , rson);
         PushUp(rt);
     }
     int query(int L,int R,int l,int r,int rt)
     {
         if (L <= l && r <= R)
         {
             return sum[rt];
         }
         PushDown(rt , r - l + );
         int m = (l + r) >> ;
         int ret = ;
         if (L <= m) ret += query(L , R , lson);
         if (m < R) ret += query(L , R , rson);
         PushUp(rt);
         return ret;
     }
 } Seg,seg;
 int main()
 {
     int T , n , m, a , b , c;
     prime();
     scanf("%d",&T);
     for (int cas =  ; cas <= T ; cas ++)
     {
         Seg.init();
         seg.init();
         scanf("%d%d",&n,&m);
         Seg.build( , n , );
         seg.build( , n , );
         repu(i,,n+)
         {
             scanf("%d",&a);
             Seg.Update(i,a,,n,);
             if(!p[a])///如果是质数才加1
                 seg.Update(i,,,n,);
         }
         char s[];
         while(m--)
         {
             scanf("%s",s);
             if(s[] == 'R')
             {
                 scanf("%d%d%d",&a, &b, &c);
                 Seg.update(b , c , a ,  , n , );
                 if(!p[a])///如果是质数，更新成1
                     seg.update(b , c ,  ,  , n , );
                 else
                     seg.update(b , c ,  ,  , n , );
             }
             else if(s[] == 'A')
             {
                 scanf("%d%d",&a,&b);///在b位置上加a，如果加的数字成了质数，也得更新另一棵树
                 int t = Seg.query(b,b,,n,);
                 Seg.Update(b,a,,n,);
                 if(!p[t+a] && p[t])///原本不是质数，变成了质数
                     seg.Update(b,,,n,);
                 else if(p[t+a] && !p[t])///原本是质数，变成了合数
                     seg.Update(b,-,,n,);
                 else if(p[t+a] && p[t]) ;
                 else if(!p[t+a] && !p[t]) ;
             }
             else
             {
                 scanf("%d%d",&a,&b);
                 printf("%d\n",seg.query(a,b,,n,));
             }
         }
     }
     return ;
 }

AC代码