zoj3886.Nico Number

线段树区间取模，维护一个最大值，如果这个取模的值大于最大值，直接返回，否则继续分段

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=401000;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int sum[maxn*4+4],maxv[maxn*4+4];
int a[maxn];
int prime[10000010];
int L,R,v;
int p;

void pushup(int rt){
    maxv[rt]=max(maxv[rt<<1],maxv[rt<<1|1]);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void build(int l,int r,int rt){
    if(l==r){
        maxv[rt]=a[l];
        if(prime[a[l]]==0)
            sum[rt]++;
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update1(int l,int r,int rt){
    if (l==r) {
        maxv[rt]=v;
        sum[rt]=!prime[maxv[rt]];
        return;
    }
    int m=(l+r)>>1;
    if(p<=m)
        update1(lson);
    else
        update1(rson);
    pushup(rt);
}

void update2(int l,int r,int rt){
    if(L<=l&&R>=r&&maxv[rt]<v){
        return ;
    }
    if (l==r) {
        maxv[rt]%=v;
        sum[rt]=!prime[maxv[rt]];
        return;
    }
    int m=(l+r)>>1;
    if(L<=m)
        update2(lson);
    if(R>m)
        update2(rson);
    pushup(rt);
}

int query(int l,int r,int rt){
    int ret=0;
    if(L<=l&&R>=r)
        return sum[rt];
    int m=(l+r)>>1;
    if(L<=m)
        ret+=query(lson);
    if(R>m)
        ret+=query(rson);
    return ret;
}

int main()
{
    int n,m;
    int op;
    memset(prime,0,sizeof(prime));
    prime[0]=0;
    prime[1]=0;
    for(int i=2;i<=10000001;i++)
        if(!prime[i]){
            if(i>10000001/i)
                continue;
            for(int j=i*i;j<10000001;j+=i)
                prime[j]=1;
    }
    prime[6]=0;
    for(int i=2;i<=10000001;i*=2)
        prime[i]=0;
    while(scanf("%d",&n)!=EOF){
        memset(sum,0,sizeof(sum));
        memset(maxv,0,sizeof(maxv));
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        build(1,n,1);
        scanf("%d",&m);
        while(m--){
            scanf("%d",&op);
            if(op==1){
                scanf("%d%d",&L,&R);
                printf("%d\n",query(1,n,1));
            }
            else if(op==2){
                scanf("%d%d%d",&L,&R,&v);
                update2(1,n,1);
            }
            else if(op==3){
                scanf("%d%d",&p,&v);
                update1(1,n,1);
            }
        }
    }
    return 0;
}