AcWing 246. 区间最大公约数（线段树）题解

原题链接

传送门：AcWing 246. 区间最大公约数

---

题目大意

有两个操作，第一个是修改：给区间[l,r]的数都加上k，第二个是查询：查询区间[l,r]内的最大公约数。

---

具体思路

这道题一看就是一个线段树的操作，而且线段树存的是区间的最大公约数（x结点的$gcd$等于左结点$gcd$与右结点$gcd$的$gcd$），这样子能很好的实现区间查询操作，但是对于区间修改很不方便，于是我们想到用线段树来存差分数组能很快的实现区间修改操作。因此我们就来想一想能不能直接用差分数组求$gcd$。
$$gcd(x_1,x_2,...,x_n)=gcd(x_1,x_2-x_1,...,x_n-x_{n-1})$$因为左边n个数的任一公约数也是右边n个数的公约数，同理右边n个数的任一公约数也是左边的公约数，因此两边的最大公约数相等。
因此我们就可以直接用差分数组来求区间最大公约数，[l,r]的最大公约数等于$gcd(x_1,tree[l,r])$，tree[l,r]存的是差分数组[l,r]区间内的$gcd$。那么对于差分数组也得用线段树存一遍，否则无法计算每个数$x_i$了，因此线段树内存差分数组和区间$gcd$即可。
注意开long long

---

AC代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 500000;
typedef long long LL;

int n,m;
LL k,w[N];
struct Node{
    int l,r;
    LL s,d;		//s存区间和，d存区间最大公约数
}tree[N*4];		//线段树开4倍空间

void pushup(Node &u,Node &l,Node &r)	//pushup操作，用左右结点跟新当前u结点
{
    u.s = l.s + r.s;		//区间和
    u.d = __gcd(l.d,r.d);	//区间最大公约数
}

void pushup(int u)	//更新当前u结点
{
    pushup(tree[u],tree[u << 1],tree[u << 1 | 1]);
}

void build(int u,int l,int r)		//建树
{
    if(l == r)		//叶子结点
        tree[u] = {l,r,w[l]-w[l-1],w[l]-w[l-1]};	//w数组求差分
    else
    {
        tree[u] = {l,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 k)	//修改操作
{
    int l = tree[u].l,r = tree[u].r;
    if(l == x && r == x)	//找到x位置
    {
        tree[u].d += k;		//加k
        tree[u].s += k;
    }
    else
    {
        int mid = l+r >> 1;
        if(x > mid)
            modify(u << 1 | 1,x,k);
        else
            modify(u << 1,x,k);
        pushup(u);
    }
}

Node query(int u,int l,int r)	//询问操作，因为要返回区间和和区间最大公约数两个变量，所以返回结构体
{
    int tl = tree[u].l,tr = tree[u].r;
    if(tl >= l && tr <= r)	//查询区间包含了当前结点，直接返回当前结点
        return tree[u];
    int mid = tl+tr >> 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);				//然后在pushup更新给res返回
        return res;
    }
}

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i ++ )
        cin >> w[i];
    build(1,1,n);	//建树
    char c;
    int l,r;
    while (m -- )
    {
        cin >> c >> l >> r;
        if(c == 'Q')		//询问操作，输出w[i]和[l+1,r]的区间gcd的gcd，注意再去绝对值
            cout << abs(__gcd(query(1,1,l).s,query(1,l+1,r).d)) << endl;
        else
        {
            cin >> k;
            modify(1,l,k);	//差分操作，给l加k
            if(r+1 <= n)	//如果r+1<=n的话再给r+1-k
                modify(1,r+1,-k);
        }
    }
    
    return 0;
}