(ACWing246)区间最大公约数(线段树+差分)

题目链接:246. 区间最大公约数 - AcWing题库

题意:给定一个长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:

  1. C l r d,表示把 A[l],A[l+1],…,A[r] 都加上 d。
  2. Q l r,表示询问 A[l],A[l+1],…,A[r]的最大公约数(GCD)。

给出一组样例:

输入样例:

5 5
1 3 5 7 9
Q 1 5
C 1 5 1
Q 1 5
C 3 3 6
Q 2 4

输出样例:

1
2
4

这道题的思路还是挺难想的,我们不可能直接维护原数组的最大公约数,因为他不支持区间修改,只支持区间查询,所以这样是不行的,我们先来看一下最大公约数的一些性质:

由辗转相减法我们可以知道(a,b)=(a,a-b)=(a-b,b),其实也容易扩展到(a,b,c)=(a,b-a,c-b),这些都是容易证明的,那看到这我们就可以往差分方向思考了,那为什么要想到差分呢?因为对于区间修改我们可以通过差分转化为单点修改,而维护最大公约数的线段树支持单调修改,所以我们要求l~r上的最大公约数,就等价于求(a[l],a[l+1]-a[l],a[l+2]-a[l+1],……,a[r]-a[r-1]),也就是求(a[l],(a[l+1]-a[l],a[l+2]-a[l+1],……,a[r]-a[r-1]))至于a[l]就是通过对差分数组求前缀和得到的,这个是比较好求的,然后剩下的就是维护一个差分数组线段树,然后对于每次修改只需要修改l和r-1两个端点,这样就可以解决这道题目了。

但是有一些细节需要注意,比如我们的差分数组可能会出现负值,但是这不影响我们求最大公约数,我们只需要把最后求出来的最大公约数取一个绝对值就好,但是千万不能直接对差分数组求绝对值,因为那样相当于我们修改了原数组,这样肯定是会出错的,还有一个需要注意的点是由于我们用到了差分数组,所以要防止数组越界,我们可以给所有数组下标都+1.

下面是代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
const int N=3e6+10;
typedef long long ll;
int l[N],r[N];
ll d[N],sum[N],gcd[N];
void pushup(int id)
{
	sum[id]=sum[id<<1]+sum[id<<1|1];
	gcd[id]=__gcd(gcd[id<<1],gcd[id<<1|1]);
}
void build(int id,int L,int R)
{
	l[id]=L;r[id]=R;
	if(L==R)
	{
		sum[id]=d[L-1];
		gcd[id]=d[L-1];
		return ;
	}
	int mid=L+R>>1;
	build(id<<1,L,mid);
	build(id<<1|1,mid+1,R);
	pushup(id);
}
void update_point(int id,int pos,ll val)
{
	if(l[id]==r[id])
	{
		sum[id]+=val;
		gcd[id]+=val;
		return ;
	}
	int mid=l[id]+r[id]>>1;
	if(pos<=mid) update_point(id<<1,pos,val);
	else update_point(id<<1|1,pos,val);
	pushup(id);
}
ll query_sum(int id,int L,int R)
{
	if(l[id]>=L&&r[id]<=R) return sum[id];
	int mid=l[id]+r[id]>>1;
	ll ans=0;
	if(L<=mid) ans+=query_sum(id<<1,L,R);
	if(mid+1<=R) ans+=query_sum(id<<1|1,L,R);
	return ans;
}
ll query_gcd(int id,int L,int R)
{
	if(l[id]>=L&&r[id]<=R) return gcd[id];
	int mid=l[id]+r[id]>>1;
	ll ans=0;
	if(L<=mid) ans=query_gcd(id<<1,L,R);
	if(mid+1<=R)
	{
		if(ans)
			ans=__gcd(ans,query_gcd(id<<1|1,L,R));
		else
			ans=query_gcd(id<<1|1,L,R);
	}
	return ans;
}
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		scanf("%lld",&d[i]);
	for(int i=n;i>=1;i--)
		d[i]-=d[i-1];
	build(1,1,n+10);
	for(int i=1;i<=m;i++)
	{
		char op[3];
		scanf("%s",op);
		int l,r;
		ll x;
		scanf("%d%d",&l,&r);
		l++;r++;//避免越界
		if(op[0]=='Q')
		{
			ll t=query_sum(1,1,l);
			printf("%lld\n",abs(__gcd(t,query_gcd(1,l+1,r))));
		}
		else
		{
			scanf("%lld",&x);
			update_point(1,l,x);
			update_point(1,r+1,-x);
		}
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值