首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >acwing-246. 区间最大公约数(线段树+gcd)[通俗易懂]

acwing-246. 区间最大公约数(线段树+gcd)[通俗易懂]

作者头像
全栈程序员站长
发布2022-09-22 10:54:21
发布2022-09-22 10:54:21
5620
举报

大家好,又见面了,我是你们的朋友全栈君。

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

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 个整数 A[i]。

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

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

每个答案占一行。

数据范围 N≤500000,M≤100000, 1≤A[i]≤1018, |d|≤1018

代码语言:javascript
复制
输入样例:
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
代码语言:javascript
复制
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5e5 + 10;
#define lson u << 1
#define rson u << 1 | 1
struct Node{ 
   
    int l,r;
    ll gcd,sum;
}trie[N * 4];
ll a[N],d[N];
int gcd(int a,int b){ 
   
    return b == 0 ? a : gcd(b, a % b);
}
void pushup(int u){ 
   
    trie[u].sum = trie[lson].sum + trie[rson].sum;
    trie[u].gcd = gcd(trie[lson].gcd,trie[rson].gcd);
}
void build(int u,int l,int r){ 
   
    if(l == r)trie[u] = { 
   l,r,d[l],d[l]};
    else{ 
   
        trie[u] = { 
   l,r};
        int mid = (l + r) >> 1;
        build(lson,l,mid);
        build(rson,mid + 1,r);
        pushup(u);
    }
}
void modify(int u,int x,ll y){ 
   
    if(trie[u].l == x && trie[u].r == x)trie[u].sum += y,trie[u].gcd += y;
    else{ 
   
        int mid = (trie[u].l + trie[u].r) >> 1;
        if(x <= mid)modify(lson,x,y);
        else modify(rson,x,y);
        pushup(u);
    }
}
Node query(int u,int l,int r){ 
   
    if(trie[u].l >= l && trie[u].r <= r){ 
   
        Node t;
        t.sum = trie[u].sum;
        t.gcd = trie[u].gcd;
    
        return t;
    }
    int mid = (trie[u].l + trie[u].r) >> 1;
    if(r <= mid)return query(lson,l,r);
    else if(l > mid)return query(rson,l,r);
    else{ 
   
        auto left = query(lson,l,r),right = query(rson,l,r);
        Node t;
        t.sum = left.sum + right.sum;
        t.gcd = gcd(left.gcd,right.gcd);
        return t;
    }
}

int main(){ 
   
    int n,m;
    cin>>n>>m;
    ios::sync_with_stdio(false);
    for(int i = 1;i <= n;i ++){ 
   
        cin>>a[i];
        d[i] = a[i] - a[i - 1];
    }
    build(1,1,n + 1);
    char t;
    ll x,y,dd;
    for(int i = 0;i < m;i ++){ 
   
        cin>>t>>x>>y;
        if(t == 'C')cin>>dd;
        if(x > y)swap(x,y);
        if(t == 'C'){ 
   
            modify(1,x,dd);
            modify(1,y + 1,-dd);
        }
        else{ 
   
            auto left = query(1,1,x),right = query(1,x + 1,y);
            cout<<gcd(left.sum,right.gcd)<<endl;
        }
    }
    return 0;
}

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/168603.html原文链接:https://javaforall.cn

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档