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;
}