CodeForces - 914D - Bash and a Tough Math Puzzle（ 线段树 + 思维 ）

题目链接：点击进入

题目

题意

n 个数，q 次操作：
1 l r x 猜测区间 [ l , r ] gcd 为 x ，若是区间内在最多修改 ( 虚拟修改 )一个值的情况下可以使区间 gcd 变为 x ，就认为这个猜测是正确的，输出 YES ，否则输出 NO；
2 i y 将 a [ i ] 改为 y ；

思路

区间查询 gcd ，单点修改，基本定型线段树
第一个操作，问的是最多修改一次的前提下，猜测的gcd对不对。
若是区间 gcd 是 x 的倍数，那么一定可以在一次修改内将区间 gcd 变为 x ；
若是不是 x 的倍数，那么若是除去一个数，剩下的数的 gcd 是 x 的倍数也行。但是这不好统计。那么我们看一下，其实若剩下的数的 gcd 是 x 的倍数，那么就说明剩下的数都是 x 的倍数。
也就是说，区间内最多只有一个数不是 x 的倍数，那这样才能在最多修改一次的前提下，认为猜测的是对的，否则就是错的。
统计区间内有多少个数不是 x 的倍数，可以用线段树解决。
对于每个节点统计对于区间 gcd ，若是 gcd 是 x 的倍数说明对应区间内都是 x 的倍数；
若是不是，说明这个区间内至少有一个不是 x 的倍数，那么继续向下查询，一直到叶子节点，若不是 x 倍数返回 1 即可。
（ 如果到这里就结束了，那么你会发现会T ）
T 掉了，我们就需要优化。其实我们刚刚提到了一个结论，即一个区间内 gcd 不是 x 的倍数那么说明区间内至少有一个数不是 x 的倍数。那么对于一个节点，我们知道它 gcd 不是 x 的倍数，那么我们可以再看一下他的左右孩子，如果它的左右孩子的 gcd 都不是 x 的倍数，说明这个节点对应区间至少有两个数不是 x 的倍数，>1 就不符合条件了，那我们直接返回 2 即可，若是只有一个孩子 gcd 不是 x 倍数，那么继续在这个孩子节点上向下查询即可，一直到叶子节点，若不是 x 倍数返回 1 即可。
第二个操作单点修改

代码

// Problem: Bash and a Tough Math Puzzle
// Contest: Virtual Judge - CodeForces
// URL: https://vjudge.net/problem/CodeForces-914D
// Memory Limit: 262 MB
// Time Limit: 2500 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=1e6+10;
const int mod=1e9+7;
const double eps=1e-9;
const int N=5e3+10;
/*--------------------------------------------*/
inline int read()
{
    int k = 0, f = 1 ;
    char c = getchar() ;
    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
    return k * f ;
}
/*--------------------------------------------*/

int n,a[maxn],q;
struct node
{
	int l;
	int r;
	int data;
}t[maxn<<2];

int gcd(int a,int b)
{
	if(!b) return a;
	else return gcd(b,a%b);
}
void pushup(int p)
{
	t[p].data=gcd(t[p<<1].data,t[p<<1|1].data);
}
void build(int p,int l,int r)
{
	t[p].l=l,t[p].r=r;
	if(l==r)
	{
		t[p].data=a[l];
		return ;
	}
	int mid=l+r>>1;
	build(p<<1,l,mid);
	build(p<<1|1,mid+1,r);
	pushup(p);
}
void change(int p,int pos,int x)
{
	if(t[p].l==t[p].r)
	{
		t[p].data=x;
		return ;
	}
	int mid=t[p].l+t[p].r>>1;
	if(pos<=mid) change(p<<1,pos,x);
	else change(p<<1|1,pos,x);
	pushup(p);
}
int ask(int p,int l,int r,int x)
{
	// if(l<=t[p].l&&t[p].r<=r&&(t[p].data%x==0))//注释部分超时
		// return 0;
	// else if(t[p].l==t[p].r)
		// return 1;
	if(l<=t[p].l&&t[p].r<=r)
	{
		if(t[p].data%x==0)
			return 0;
		else if(t[p].l==t[p].r)
			return 1;
		else if((t[p<<1].data%x)&&(t[p<<1|1].data%x))
			return 2;
		else if(t[p<<1].data%x)
			return ask(p<<1,l,r,x);
		else if(t[p<<1|1].data%x)
			return ask(p<<1|1,l,r,x);
	}
	int mid=t[p].l+t[p].r>>1,ans=0;
	if(l<=mid) ans+=ask(p<<1,l,r,x);
	if(r>mid) ans+=ask(p<<1|1,l,r,x);
	return ans;
}

int main() 
{
	// ios::sync_with_stdio(false);
	// cin.tie(0);cout.tie(0);
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	build(1,1,n);
	scanf("%d",&q);
	while(q--)
	{
		int op,x,y,z;
		scanf("%d%d%d",&op,&x,&y);
		if(op==1)
		{
			scanf("%d",&z);
			int sum=ask(1,x,y,z);
			if(sum<=1) printf("YES\n");
			else printf("NO\n");
		}
		else
			change(1,x,y);
	}
    return 0;
}