Interval GCD 题解

Interval GCD 题解

题面：

        给定一个长度为N的数列A，以及M条指令 (N≤5*10^5, M<=10^5)，每条指令可能是以下两种之一：

        “C l r d”，表示把 A[l],A[l+1],…,A[r] 都加上 d。

        “Q l r”，表示询问 A[l],A[l+1],…,A[r] 的最大公约数(GCD)。

输入：

        第一行两个整数N,M，第二行N个整数Ai，接下来M行每条指令的格式如题目描述所示。

输出：

        对于每个询问，输出一个整数表示答案。

样例输入：

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

提示：

        N,M≤2*10^5，l<=r，数据保证任何时刻序列中的数都是不超过2^62-1的正整数。

简要题意：

         对一个数列进行2种操作，把A[l],A[l+1],…,A[r]都加上d或询问A[l],A[l+1],…,A[r]的最大公约数(GCD)并输出这个数。

解题思路：

        首先，看到有关区间增减操作，就想到了线段树和树状数组，在前面的题目（不知道线段树和树状数组的先看文章（线段树）（线段树模板）和（树状数组）（树状数组模板））中就已经解决了这种问题，这道题的一个新的考点是关于求最大公因数。如果一个一个求时间复杂度是不行的，就自然而然地想到了求最大公因数的数学方法：辗转相除法。

        这个方法用字母表示就是gcd(x,y)=gcd(x,y-x)，我们可以扩大成三元的：gcd(x,y,z)=gcd(x,y-x,z-y),再扩大成n元的：gcd(a1,a2,a3,a4,...,an)=gcd(a1,a2-a1,a3-a2,...,an-a(n-1));

        不难发现其实这就是一个差分数组，所以我们只要用树状数组和线段树来维护这样一个差分序列，每次更新就不用l~r全部更新，只需要更新头和尾，然后通过getsum和query函数找到最大公因数即可。

完整代码：

#include<bits/stdc++.h>
using namespace std;

#define lson k<<1,l,mid
#define rson k<<1|1,mid+1,r
#define ll long long

int n,m;
ll s[2500000];
ll a[2500000];
ll c[2500000];

int lowbit(int i){
	return i&-i;
}

void update(int x,ll y){//树状数组改变数值
	for(int i=x;i<=n;i+=lowbit(i)){
		c[i]+=y;
	}
}

ll getsum(int x){//树状数组获取数值
    ll val=0;
    for(int i=x;i!=0;i-=lowbit(i)){ 
		val+=c[i];	
	}
    return val;
}

void pushup(int k){//线段树更新操作
	s[k]=__gcd(s[k<<1],s[k<<1|1]);//求最大公因数
}

void tree(int k,int l,int r){//建树部分
	if(l==r){
		s[k]=a[l]-a[l-1];//差分操作
		return;
	}
	int mid=(l+r)>>1;
	tree(lson);
	tree(rson);
	pushup(k);
}

void change(int k,int l,int r,int x,ll y){//线段树更新部分
	if(l==r){
		s[k]+=y;//数值增加
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid){
		change(lson,x,y);
	}
	else{
		change(rson,x,y);
	}
	pushup(k);
}

ll query(int k,int l,int r,int x,int y){//线段树读取数值函数
	if(x<=l&&r<=y){
		return s[k];
	}
	int mid=(l+r)>>1;
	if(y<=mid){
		return query(lson,x,y);
	}
	else{
		if(x>mid){
			return query(rson,x,y);
		}
		else{
			return __gcd(query(lson,x,y),query(rson,x,y));
		}
	}
}

int main(){
	int x,y;
	ll z;
	char C[10];
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
		update(i,a[i]-a[i-1]);//进行差分操作
	}
	tree(1,1,n);
	for(int i=1;i<=m;i++){
		scanf("%s%d%d",&C,&x,&y);
		if(C[0]=='C'){
			scanf("%lld",&z);
			update(x,z);
			update(y+1,-z);//树状数组差分维护
			change(1,1,n,x,z);//线段树差分维护
			if(y+1<=n){//若y超出了n的范围，则不用更新
				change(1,1,n,y+1,-z);
			}
		}
		else{
			if(x==y){//如果相等，直接输出1
				printf("1\n");
			}
			else{
				printf("%lld\n",abs(__gcd(getsum(x),query(1,1,n,x+1,y))));//输出最大公因数（差分可能为负数，所以要abs取绝对值） 
			}
		}
	}
	return 0;
}