【cf914D】Bash and a Tough Math Puzzle 线段树

题目大意

给出一个序列a，要求支持单点修改，以及询问如果允许修改区间内的一个元素，区间gcd是否为x（可以理解为允许你暂时去掉区间一个元素，使区间gcd为x）
1 ≤ n ≤ 5*10^5 1 ≤ q ≤ 4·10^5

题解

一开始看题很萌比 后来看了题解发现自己是傻比系列
我们开一个线段树 每个节点记录当前区间的gcd，每次询问只要查一下当前区间的gcd是否为x如果不是就继续往下找，记录一下需要修改几次，只要找到需要修改两个叶子节点就不要找了，返回NO。否则返回YES

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cctype>
#define N 500010
#define lson l,mid,st<<1
#define rson mid+1,r,st<<1|1
using namespace std;
int tree[4*N],n,m;
int gcd(int a,int b){while(b^=a^=b^=a%=b);return a;}
void pushup(int x){tree[x]=gcd(tree[x<<1],tree[x<<1|1]);}
void build(int l,int r,int st)
{
	if (l==r) {scanf("%d",&tree[st]);return;}
	int mid=(l+r)>>1;
	build(lson);build(rson);
	pushup(st);
}
void update(int x,int d,int l,int r,int st)
{
	if (l==r) {tree[st]=d;return;}
	int mid=(l+r)>>1;
	if (x<=mid) update(x,d,lson);
	else update(x,d,rson);
	pushup(st);
}
int query(int L,int R,int x,int l,int r,int st)
{
	if (l==r) return 1;
	int mid=(l+r)>>1,ret=0;
	if (L<=mid&&tree[st<<1]%x!=0) ret+=query(L,R,x,lson);
	if (R>mid&&tree[st<<1|1]%x!=0&&ret<=1) ret+=query(L,R,x,rson);
	return ret;  
}
int main()
{
	scanf("%d",&n);
	build(1,n,1);
	scanf("%d",&m);
	for (int i=1;i<=m;i++)
	{
		int op,L,R,x;
		scanf("%d",&op);
		if (op==2) 
			scanf("%d%d",&L,&R),update(L,R,1,n,1);
		else 
			scanf("%d%d%d",&L,&R,&x),query(L,R,x,1,n,1)<=1?printf("YES\n"):printf("NO\n");
	}
	return 0;
}