Codecraft-18 and Codeforces Round #458 D. Bash and a Tough Math Puzzle（线段树）

题目链接：点击打开链接

题目大意：给出一个数列a：a1，a2，a3......an

有两种询问：

第一种：1 l r x  

意思是在数列a的[l，r]的区间中，如果在最多改变一个数字的情况（不是真的改变，只是假设改变的话），这些数字的公约数能被x整除则输出YES，否则输出NO。

第二种：2 i y

将数列a中的ai的值改为y

思路：

典型的线段树的题，单点更新。

首先叶子结点记录的就是这个数列，然后父节点是他两个左右子节点的最大公约数。

注意在查询过程中，需要记录的是在查询的的区间内有多少叶子结点中不含约数x

在更新过程中，自下往上更新即可。

以下给出样例2的线段树初始状态。

代码如下：

#include <bits/stdc++.h>
using namespace std;
#define LL long long
#define maxn 500000
int tree[maxn<<2];
int rt=1;
int gcd(int x,int y)
{
    return y==0?x:gcd(y,x%y);
}
void query(int root,int L,int R,int x,int l,int r,int &ans)
{
    if(r<l||R<L||r<L||R<l)
    {
        return ;
    }
    else if(L<=l&&r<=R)
    {
        if(tree[root]%x==0)
        {
            return ;
        }
        else
        {
            while(root<rt)
            {
                if(tree[2*root]%x!=0)
                {
                    if(tree[2*root+1]%x!=0)//左右两个子树的最大公约数都不能整除x  不能整除的数至少有两个
                    {
                        ans+=2;
                        return ;
                    }
                    root=2*root;
                }
                else
                {
                    root=2*root+1;
                }
            }
            ans++;//有一个数不能被x整除
            return ;
        }
    }
    int mid=(l+r)/2;
    query(2*root,L,R,x,l,mid,ans);
    if(ans>1)
    {
        return ;
    }
    query(2*root+1,L,R,x,mid+1,r,ans);
}
void update(int p,int y)
{
    p+=rt-1;
    tree[p]=y;

    p=p/2;
    while(p>=1)
    {
        tree[p]=gcd(tree[2*p],tree[2*p+1]);
        p=p/2;
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        while(rt<n)
        {
            rt<<=1;
        }
        for(int i=rt;i<rt+n;i++)
        {
            scanf("%d",&tree[i]);
        }
        for(int i=rt-1;i>=1;i--)
        {
            tree[i]=gcd(tree[2*i],tree[2*i+1]);
        }
        int q;
        scanf("%d",&q);
        while(q--)
        {
            int t;
            scanf("%d",&t);
            if(t==1)
            {
                int l,r,x;
                scanf("%d%d%d",&l,&r,&x);
                int ans=0;
                query(1,l,r,x,1,rt,ans);
                if(ans>1)
                {
                    printf("NO\n");
                }
                else
                {
                    printf("YES\n");
                }
            }
            else
            {
                int i,y;
                scanf("%d%d",&i,&y);
                update(i,y);
            }
        }
    }
    return 0;
}