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

题目链接

题意

给出一段序列 $a[1...n]$，进行以下操作

1. 单点修改
2. 查询能否至多修改一个数，使区间 $[L,R]$ 的GCD为 $x$ 的倍数

题解

线段树维护区间GCD，查询的时候一共有 3 种情况

1. 左子树 GCD%x=0， 右子树 GCD%x=0，此时答案为YES
2. 左右子树其中一颗 GDD%x=0，另一颗 GCD%x!=0，递归那颗不为 GCD%x!=0 的
3. 左右子树 GCD%x 都不为0，答案为NO，因为左右子树至少都需要修改一次

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<list>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#include<ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
#include<ext/rope>
using namespace __gnu_cxx;

#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))

const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
const int INF = 0x3f3f3f3f;
const int N = 5e6+10;
struct node{
	int l,r,mid;
	int w; 
}seg[N];
int a[N];
void pu(int rt){seg[rt].w=__gcd(seg[rt<<1].w,seg[rt<<1|1].w);}
void build(int rt,int l,int r){
	int mid=l+r>>1;
	seg[rt].l=l,seg[rt].r=r,seg[rt].mid=mid;
	if(l==r){seg[rt].w=a[l]; return ;}
	build(rt<<1,l,mid),build(rt<<1|1,mid+1,r);
	pu(rt);
}
void update(int rt,int pos,int val){
	if(seg[rt].l==seg[rt].r){seg[rt].w=val;return;}
	if(pos<=seg[rt].mid)update(rt<<1,pos,val);
	else update(rt<<1|1,pos,val);
	pu(rt);
} 
int query(int rt,int l,int r){
	if(seg[rt].r<l||seg[rt].l>r)return 0;
	if(l<=seg[rt].l&&seg[rt].r<=r)return seg[rt].w;
	int ans=0;
	if(l<=seg[rt].mid)ans=__gcd(ans,query(rt<<1,l,r));
	if(seg[rt].mid<r)ans=__gcd(ans,query(rt<<1|1,l,r));
	return ans;
}
bool check(int rt,int l,int r,int x){
	if(l==r) return 1;
	int ansl=x,ansr=x;
    if(seg[rt].mid>=l) ansl=query(rt<<1,l,seg[rt].mid);
    if(seg[rt].mid<r) ansr=query(rt<<1|1,seg[rt].mid+1,r);
    ansl=__gcd(ansl,x);
    ansr=__gcd(ansr,x);
    if(ansl!=x&&ansr!=x) return 0;
    else if(ansl==x&&ansr==x) return 1;
    else if(ansl==x) return check(rt<<1|1,seg[rt].mid+1,r,x);
    else return check(rt<<1,l,seg[rt].mid,x);
}
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	int n; cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	build(1,1,n);
	int q; cin>>q;
	while(q--){
		int op,x,y,z; cin>>op>>x>>y;
		if(op==1){
			cin>>z;
			if(check(1,x,y,z))cout<<"YES"<<endl;
			else cout<<"NO"<<endl;
		}
		if(op==2) update(1,x,y);
	}
	
}