线段树(求区间最大公约数)

q619718

已于 2023-02-13 20:15:48 修改

阅读量137

点赞数

文章标签：算法数据结构图论

于 2023-02-04 11:43:11 首次发布

本文链接：https://blog.csdn.net/q619718/article/details/128880219

版权

该文章描述了一个程序设计问题，给定一个数列和一系列指令，包括对数列部分元素的加法操作以及查询连续子序列的最大公约数。解决方案使用了二叉树结构来高效处理这些操作，其中二叉树节点包含子序列的和与最大公约数信息，支持快速查询和更新。

摘要由CSDN通过智能技术生成

给定一个长度为 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

输入样例：
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

#include <iostream>

using namespace std;
typedef long long ll;
constexpr int N=500010;
int n,m;
ll w[N];
struct node{
    int l,r;
    ll sum,d;
}tr[N*4];
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(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|1,mid+1,r);
        build(u<<1,l,mid);
        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(l>r) return {0};
    if(tr[u].l>=l&&tr[u].r<=r) return tr[u];
    else{
        int mid=tr[u].l+tr[u].r>>1;
        if(l>mid) return query(u<<1|1,l,r);
        else if (r<=mid)return query(u<<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()
{
    scanf("%d%d",&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--)
    {
        scanf("%s%d%d",op,&l,&r);
        if(*op=='Q')
        {
            auto left=query(1,1,l);
            auto right=query(1,l+1,r);
            printf("%lld\n",abs(gcd(left.sum,right.d)));
        }
        else
        {
            scanf("%lld",&d);
            modify(1,l,d);
            if(r+1<=n) modify(1,r+1,-d);
        }
    }

    return 0;
}