【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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值