[Acwing] 区间最大公约数 线段树

前言

传送门:
总体思路很流畅,就是代码不怎么好写

思路

题目要求两个操作 :

• 区间更新 : 懒标记 (难写啊！
• 区间查询 : 查询$gcd$更不好写!

区间更新的优化 :

区间$+$ 或者 $-$ 一个数 可以用 差分 变成单点操作

区间查询优化 :

根据 $gcd(a,b)=gcd(a,b-a).....gcd(a,b,c,d...)=gcd(a,b-a,c-b,...)$

公式请拐弯到y总视频

因此我们只需要特殊处理 第一个点 和 后面用 差分处理 即可

CODE

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define ned '\n'
const int N = 5e5+10;
int n,m;
ll w[N];

struct node
{
    int l,r;
    ll sum,d;
} tr[N*4];
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(tr[u],tr[u<<1],tr[u<<1|1]);
}

void build(int u,int l,int r)
{
    if(l == r)
    {
        ll b = w[r] -w[r-1];
        tr[u] = {l,r,b,b};
    }
    else
    {
        tr[u].l  =l ;
        tr[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(tr[u].l == x && tr[u].r == x)
    {
        ll b = tr[u].sum + v;
        tr[u] = {x,x,b,b};
    }else
    {
        int mid = tr[u].l+tr[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(tr[u].l >=l  && tr[u].r <=r ) return tr[u];
    else
    {
        int mid = tr[u].l + tr[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;

        }
    }
}
void solve()
{
    cin>>n>>m;
    for(int i=1; i<=n; i++)
        cin>>w[i];

    build(1,1,n);

    int l,r;
    ll d;
    char op[2];

    while(m -- )
    {
        cin>>op>>l>>r;



        if(*op == 'Q')
        {
            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))<<ned;
            
        }
        else
        {
            cin>>d;
            modify(1,l,d);
            if(r+1<=n)
            modify(1,r+1,-d);
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    solve();
    return 0;
}