SPOJ CNTPRIME 13015 Counting Primes （水题，区间更新，求区间的素数个数）

题目连接：http://www.spoj.com/problems/CNTPRIME/

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define lson rt<<1,L,mid
#define rson rt<<1|1,mid+1,R
/*
水题
题意：给出n个初始值，给出两种操作
      0 x y v：将[x,y]的数改成v
      1 x y ：查询[x,y]区间有多少个素数（相同的数，有几个算几个，即有两个3，那么个数为2）
*/
using namespace std;
const int maxn=;
int t,n,q;
bool isprime[maxn];

struct Node{
    int sum;  //统计该区间的素数个数
    int lazy;
    int len;
}tree[maxn<<];

//素数筛选法
void init(){
    memset(isprime,true,sizeof(isprime));
    for(int i=;i<maxn;i++){
        if(isprime[i]){
            for(int j=i*;j<maxn;j+=i){
                isprime[j]=false;
            }
        }
    }
}
void pushUp(int rt){
    tree[rt].sum=tree[rt<<].sum+tree[rt<<|].sum;
}
void pushDown(int rt){
    if(tree[rt].lazy!=-){
        tree[rt<<].lazy=tree[rt<<|].lazy=tree[rt].lazy;
        tree[rt<<].sum=tree[rt].lazy?tree[rt<<].len:;
        tree[rt<<|].sum=tree[rt].lazy?tree[rt<<|].len:;
        tree[rt].lazy=-;
    }
}
void build(int rt,int L,int R){
    tree[rt].lazy=-;
    tree[rt].len=R-L+;
    if(L==R){
        int v;
        scanf("%d",&v);
        if(isprime[v])
            tree[rt].sum=;
        else
            tree[rt].sum=;
        return;
    }
    int mid=(L+R)>>;
    build(lson);
    build(rson);
    pushUp(rt);
}
void update(int rt,int L,int R,int l,int r,int v){
    if(l<=L&&R<=r){
        if(isprime[v]){
            tree[rt].sum=R-L+;
            tree[rt].lazy=;
        }
        else{
            tree[rt].sum=;
            tree[rt].lazy=;
        }
        return;
    }
    pushDown(rt);
    int mid=(L+R)>>;
    if(l<=mid)
        update(lson,l,r,v);
    if(r>mid)
        update(rson,l,r,v);
    pushUp(rt);
}
int query(int rt,int L,int R,int l,int r){
    if(l<=L&&R<=r){
        return tree[rt].sum;
    }
    int mid=(R+L)>>;
    int ret=;
    pushDown(rt);
    if(l<=mid)
        ret+=query(lson,l,r);
    if(r>mid)
        ret+=query(rson,l,r);
    return ret;
}
int main()
{
    init();
    int op,x,y,v;
    scanf("%d",&t);
    for(int i=;i<=t;i++){
        printf("Case %d:\n",i);
        scanf("%d%d",&n,&q);
        build(,,n);
        for(int j=;j<=q;j++){
            scanf("%d",&op);
            if(op==){
                scanf("%d%d%d",&x,&y,&v);
                update(,,n,x,y,v);
            }
            else{
                scanf("%d%d",&x,&y);
                printf("%d\n",query(,,n,x,y));
            }
        }
    }
    return ;
}