ZOJ 3911Prime Query [素数处理 + 线段树]

Time Limit: 5 Seconds Memory Limit: 196608 KB
You are given a simple task. Given a sequence A[i] with N numbers. You have to perform Q operations on the given sequence. 
Here are the operations: 
* A v l, add the value v to element with index l.(1<=V<=1000) 
* R a l r, replace all the elements of sequence with index i(l<=i<= r) with a(1<=a<=10^6) 
* Q l r, print the number of elements with index i(l<=i<=r) and A[i] is a prime number 
Note that no number in sequence ever will exceed 10^7.

Input 
The first line is a signer integer T which is the number of test cases. 
For each test case, The first line contains two numbers N and Q (1 <= N, Q <= 100000) - the number of elements in sequence and the number of queries. 
The second line contains N numbers - the elements of the sequence. 
In next Q lines, each line contains an operation to be performed on the sequence.

Output 
For each test case and each query,print the answer in one line.

Sample Input 
1 
5 10 
1 2 3 4 5 
A 3 1 
Q 1 3 
R 5 2 4 
A 1 1 
Q 1 1 
Q 1 2 
Q 1 4 
A 3 5 
Q 5 5 
Q 1 5

Sample Output 
2 
1 
2 
4 
0 
4

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#define INF 0x3f3f3f3f
#define EPS 0.00000001
#define lowbit(x) (x&(-x))
using namespace std;
typedef long long ll;
typedef unsigned long long ull;

const int maxn = 1e5+;
const int N = 1e7+;
int t[maxn << ],in[maxn << ],lazy[maxn << ];
int prime[N];

void getprime()
{
    for(int i=;i<sqrt(1.0*N);i++)
        if(!prime[i])
            for(int j=i+i;j<N;j+=i)
                prime[j] = ;
    prime[] = ;
    prime[] = ;
}

void pushdown(int i,int b,int e)
{
    if(lazy[i])
    {
        int mid = (b + e) / ;
        in[i << ] = !prime[lazy[i]] * (mid - b + );
        in[i <<  | ] = !prime[lazy[i]] * (e - mid);

        t[i << ] = lazy[i];
        t[i <<  | ] = lazy[i];

        lazy[i << ] = lazy[i];
        lazy[i <<  | ] = lazy[i];

        lazy[i] = ;
    }
}

void add(int i,int b,int e,int pos,int val)
{
    if(b == e)
    {
        in[i] -= !prime[t[i]];
        t[i] += val;
        in[i] += !prime[t[i]];
        return ;
    }
    pushdown(i, b, e);

    int mid = (b + e) / ;
    if(pos <= mid) add(i << , b, mid, pos, val);
    else add(i <<  | , mid + , e, pos, val);

    in[i] = in[i << ] + in[i <<  | ];
}

void replace(int i,int b,int e,int l,int r,int val)
{
    if(b >= l && e <= r)
    {
        in[i] = (!prime[val]) * (e - b + );
        t[i] = val;
        lazy[i] = val;
        return ;
    }
    pushdown(i, b, e);

    int mid = (b + e) / ;
    if(r <= mid) replace(i << , b, mid, l, r, val);
    else if(l > mid) replace(i<<  | , mid + , e, l, r, val);
    else
    {
        replace(i << , b, mid, l, r, val);
        replace(i <<  | , mid + , e, l, r, val);
    }

    in[i] = in[i << ] + in[i <<  | ];
}

int query(int i,int b,int e,int l,int r)
{
    if(b >= l && e <= r)
        return in[i];
    pushdown(i, b, e);

    int mid = (b + e) / ;
    if(r <= mid) return query(i << , b, mid, l, r);
    else if(l > mid) return query(i <<  | , mid + , e, l, r);
    else return query(i << , b, mid, l, r) + query(i <<  | , mid + , e, l, r);
}

int main()
{
    getprime();

    int T;
    scanf("%d",&T);
    while(T--)
    {
        char s[];
        int n,m,x,a,b,c;

        scanf("%d%d",&n,&m);

        memset(t,,sizeof(t));
        memset(in,,sizeof(in));
        memset(lazy,,sizeof(lazy));

        for(int i=;i<=n;i++)
        {
            scanf("%d",&x);
            add(,,n,i,x);
        }

        for(int i=;i<m;i++)
        {
            scanf("%s%d%d",s,&a,&b);
            if(s[] == 'A') add(,,n,b,a);

            else if(s[] == 'Q') printf("%d\n", query(,,n,a,b));

            else
            {
                scanf("%d",&c);
                replace(,,n,b,c,a);
            }

        }
    }
}