zoj3886.Nico Number

7875 ワード

セグメントツリー区間で型を取り、最大値を維持します.この型の値が最大値より大きい場合は、直接戻ります.そうしないと、セグメント化が続行されます.
#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
"
,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; }