cf-914D-线段树

http://codeforces.com/contest/914/problem/D

 题目大意是给出一个数列，进行两种操作，一个是将位置i的数置为x，另一个操作是询问[l,r]内的数的gcd是不是x，我们可以至多更改这个区间内的一个数。

线段树维护区间gcd，询问的时候统计有多少个数不是x的倍数，当数量大于1的时候就可以直接return了，因为答案显然已经是NO了，没这一句的话会T。

　　

#include<bits/stdc++.h>
using namespace std;
#define LL long long  
#define ULL unsigned long long 
const int maxn=500050;
int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}
struct sg_tree{
    #define mid ((L+R)>>1)
    #define lc (id<<1)
    #define rc (id<<1|1)

    int g[maxn<<2],N;
    void build(int id,int L,int R){
        if(L==R){
            scanf("%d",&g[id]);
            return;
        }
        build(lc,L,mid);
        build(rc,mid+1,R);
        g[id]=gcd(g[lc],g[rc]);
    }

    void change(int id,int L,int R,int tar,int v){
        if(L==R){
            g[id]=v;
            return;
        }
        if(tar<=mid) change(lc,L,mid,tar,v);
        else change(rc,mid+1,R,tar,v);
        g[id]=gcd(g[lc],g[rc]);
    }

    void ask(int id,int L,int R,int l,int r,int x,int &tot){
        if(tot>1) return; 
        if(L==R){
            if(g[id]%x!=0) tot++;
            return;
        }
        if(l<=mid){
            if(g[lc]%x!=0){
                ask(lc,L,mid,l,r,x,tot);
            }
        }
        if(r>mid){
            if(g[rc]%x!=0){
                ask(rc,mid+1,R,l,r,x,tot);
            }
        }
    }
}ac;
int main(){
    scanf("%d",&ac.N);
    ac.build(1,1,ac.N);
    int op,m,l,r,x,i;
    scanf("%d",&m);
    while(m--){
        scanf("%d",&op);
        if(op==1){
            int ok=0;
            scanf("%d%d%d",&l,&r,&x);
            ac.ask(1,1,ac.N,l,r,x,ok);
            ok<=1?puts("YES"):puts("NO");
        }
        else{
            scanf("%d%d",&i,&x);
            ac.change(1,1,ac.N,i,x);
        }
    }
    return 0;
}