算法竞赛进阶指南 0X40数据结构进阶——区间最大公约数

题目链接

AcWing 246. 区间最大公约数 hard

题目描述

给定一个长度为 $N$ 的数列 $A$，以及 $M$ 条指令，每条指令可能是以下两种之一：

• C l r d，表示把 $A[l],A[l+1],\dots ,A[r]$ 都加上 $d$。
• Q l r，表示询问 $A[l],A[l+1],\dots ,A[r]$ 的最大公约数(GCD)。

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

输入格式

第一行两个整数 $N,M$。

第二行 $N$ 个整数 $A[i]$。

接下来 $M$ 行表示 $M$ 条指令，每条指令的格式如题目描述所示。

输出格式

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

每个答案占一行。

数据范围

• $N\le 500000,M\le 100000$

• $1\le A[i]\le 10^{18}$

• $|d|\le 10^{18}$

• 保证数据在计算过程中不会超过 long long 范围。

输入样例：

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

解法：线段树

对于 C l r d 这种在一个区间 $[l,r]$ 上加上某一个数 $d$ ，我们可以使用差分数组来做，将一次区间操作转换为两次单点操作。

我们使用 $b$ 为原数组 $a$ 的差分数组，即 $b[1]=a[1]-a[0],b[2]=a[2]-a[1],...$。

那么区间 $[l,r]$ 加上 $d$ 就等价于：

• $b[l]$ 加上 $d$;
• $b[r+1]$ 减去 $d$；

所以线段树 $t$ 就可以维护这个差分数组 $b$，对于这个数组的前 $x$ 项和就是 $a[x]$。

对于 Q l r 这个求区间的最大公约数，直接遍历求，时间复杂度我们是不能接受的。

所以我们考虑将这个区间操作，也转换为类似的单点操作。

我们可以知道 GCD 有如下性质：

• $gcd(a,b)=g(a,b-a)$
• $gcd(a,b,c)=g(a,b-a,c-b)$
• …
• $gcd(a_1,a_2,a_3,a_4,...,a_{n-1},a_n)=g(a_1,a_2-a_1,a_3-a_2,a_4-a_3,...,a_{n-1}-a_{n-2},a_n-a_{n-1})$

所以我们也可以维护一个 $gcd$ 的差分数组，并且这个差分数组就等同于之前的这个差分数组 $b$。

每一个节点，我们用 $sum$ 来维护当前节点区间 $[t[u].l,t[u].r]$ 的前缀和 ； 用 $d$ 来表示当前节点区间 $[t[u].l,t[u].r]$ 的最大公约数。

要求 $[l,r]$ 的最大公约数，就是 g c d { a l + g c d ( a l + 1 − a l , a l + 2 − a l + 1 , . . . , a r − a r − 1 ) } gcd \{ a_l + gcd(a_{l + 1} - a_{l} , a_{l + 2} - a_{l + 1},...,a_r - a_{r - 1}) \} gcd{al​+gcd(al+1​−al​,al+2​−al+1​,...,ar​−ar−1​)} ， 即 g c d { q u e r y ( 1 , 1 , l ) . s u m , q u e r y ( l + 1 , r ) . d } gcd \{ query(1,1,l).sum , query(l + 1 , r).d \} gcd{query(1,1,l).sum,query(l+1,r).d}。

• $query(1,1,l).sum$就是差分数组的前 $l$ 项和，也就是 $a_l$ 的值；
• $query(l+1,r).d$ 就是 $gcd(a_{l+1}-a_l,a_{l+2}-a_{l+1},...,a_r-a_{r-1})$ 的值；
• 所以两者在求 $gcd$ 就是我们的答案；

时间复杂度： $O(logn)$ 实现区间查询、单点修改、区间修改

代码：

#include<iostream>
using namespace std;

using LL = long long;

const int N = 5e5+10;

struct node{
    int l , r;
    LL sum , d;
}t[N * 4];

LL a[N];

int n,m;

LL gcd(LL a , LL b){
    return b ? gcd(b , a % b) : a;
}

void pushup(node& u,node& l,node& r){
    u.sum = l.sum + r.sum;
    u.d = gcd(l.d , r.d);
}

void pushup(int u){
    pushup(t[u] , t[u<<1] , t[u<<1|1]);
}

void build(int u,int l,int r){
    if(l == r){
        LL b = a[l] - a[l - 1];
        t[u] = {l,r,b,b};
        return;
    }
    
    t[u].l = l , t[u].r = r;
    
    int mid = (l + r) >> 1;
    
    build(u<<1 , l , mid);
    build(u<<1|1 , mid + 1 , r);
    
    pushup(u);
}

void modify(int u , int x , LL v){
    if(t[u].l == x && t[u].r  == x){
        LL b = t[u].sum + v;
        t[u] = {x,x,b,b};
        return;
    }
    
    int mid = (t[u].l + t[u].r) >> 1;
    
    if(x <= mid) modify(u << 1 , x , v);
    else modify(u << 1 | 1 , x , v);
    
    pushup(u);
}

node query(int u,int l ,int r){
    if(l <= t[u].l && t[u].r <= r){
        return t[u];
    }
    
    int mid = (t[u].l + t[u].r) >> 1;
    
    if(r <= mid) return query(u << 1 , l  ,r);
    else if(l > mid) return query(u << 1 | 1  , l ,r);
    else{
        auto left = query(u << 1 , l  ,r);
        auto right = query(u << 1 | 1  , l ,r);
        node res;
        pushup(res,left,right);
        return res;
    }
} 

int main(){
    cin>>n>>m;
    
    for(int i = 1;i <= n;++i) scanf("%lld",&a[i]);
    
    //建树
    build(1,1,n);
    
    int l , r;
    char ch;
    while(m --){
        cin>>ch>>l>>r;
        
        if(ch == 'C'){
            LL d;
            cin>>d;
            modify(1,l,d);
            if(r + 1 <= n) modify(1,r + 1 , -d);
        }
        else{
            auto left = query(1,1,l);
            node right = {0,0,0,0};
            if(l + 1 <= r) right = query(1,l + 1 , r);
            cout<<abs(gcd(left.sum  , right.d))<<'\n';
        }
    }
    return 0;
}