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;
}