牛客算法周周练8C 小阳的贝壳 (线段树+差分数组)

链接：https://ac.nowcoder.com/acm/contest/5803/C
来源：牛客网

题目描述

小阳手中一共有 n 个贝壳，每个贝壳都有颜色，且初始第 i 个贝壳的颜色为 colicol_icoli​ 。现在小阳有 3 种操作：

1 l r x：给 [l,r] 区间里所有贝壳的颜色值加上 x 。

2 l r：询问 [l,r] 区间里所有相邻贝壳 颜色值的差（取绝对值） 的最大值（若 l==r 输出 0）。

3 l r ：询问 [l,r] 区间里所有贝壳颜色值的最大公约数。

输入描述:

第一行输入两个正整数 n,m，分别表示贝壳个数和操作个数。
第二行输入n 个数 col_i，表示每个贝壳的初始颜色。
第三到第 m++2 行，每行第一个数为 opt，表示操作编号。接下来的输入的变量与操作编号对应。

输出描述:

共 m 行，对于每个询问（操作 2 和操作 3）输出对应的结果。

示例1

输入
复制

5 6
2 2 3 3 3
1 2 3 3
2 2 4
3 3 5
1 1 4 2
3 2 3
2 3 5

输出
复制

3
3
1
3

备注:

$1\le n,m\le 10^5,1\le coli,x\le 10^3,1\le opt\le 3,1\le l\le r\le n$

分析

破题的关键在于gcd的一条性质
$gcd(a_1,a_2,...,a_n)=gcd(a_1,a_2-a_1,a_3-a_2,...,a_n-a_{n-1})$
因此我们可以把输入数据$(a_1,a_2,...,a_n)$处理成为一个差分数组$(a_1,a_2-a_1,a_3-a_2,...,a_n-a_{n-1})$
对于询问区间最大绝对差值，可以直接用建立在差分数组上的线段树解决
对于询问区间[l,r]最大公约数，相当于询问gcd(value[l],query_maxgcd(l+1,r))
对于区间修改，因为我们维护的是差分数组，所以相当于a[l]+k,a[r+1]-k
在这个线段树上求某个点的原始值value[i]，相当于在差分数组上计算sum(1,i)，所以也可以用线段树解决。
我写的又丑又长，放个写的简洁的大佬的

#include<bits/stdc++.h>
#define ls cur<<1
#define rs cur<<1|1
using namespace std;
 
typedef long long ll;
const int maxn=1e5+10;
 
int sum[maxn<<2],a[maxn<<2],ma[maxn<<2],gd[maxn<<2];
 
 
void pushup( int cur )
{
    gd[cur]=__gcd(gd[ls],gd[rs]);
    ma[cur]=max(ma[ls],ma[rs]);
    sum[cur]=sum[ls]+sum[rs];
}
 
void build( int cur,int l,int r )
{
    if( l==r )
    {
        sum[cur]=a[l];
        gd[cur]=ma[cur]=abs(a[l]);
        return;
    }
    int mid=l+r>>1;
    build(ls,l,mid);
    build(rs,mid+1,r);
    pushup(cur);
}
 
void update( int cur,int l,int r,int p )
{
    if( l==r )
    {
        sum[cur]=a[l];
        gd[cur]=ma[cur]=abs(a[l]);
        return;
    }
    int mid=l+r>>1;
    if( p<=mid ) update(ls,l,mid,p);
    else update(rs,mid+1,r,p);
    pushup(cur);
}
 
int get_sum( int cur,int l,int r,int L,int R )
{
    if( L<=l && r<=R ) return sum[cur];
    int mid=l+r>>1;
    int ans=0;
    if( L<=mid ) ans+=get_sum(ls,l,mid,L,R);
    if( R>mid ) ans+=get_sum(rs,mid+1,r,L,R);
    return ans;
}
 
int query( int cur,int l,int r,int L,int R, int id ) // 1 差值 / 0 为 公约数
{
    if( L<=l && r<=R ) return id ? ma[cur] : gd[cur];
    int mid=l+r>>1;
    int tmp1=0,tmp2=0;
    if( L<=mid ) tmp1=query(ls,l,mid,L,R,id);
    if( R>mid ) tmp2=query(rs,mid+1,r,L,R,id);
    return id ? max(tmp1,tmp2) : __gcd(tmp1,tmp2); 
}
 
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for( int i=1;i<=n;i++ ) scanf("%d",&a[i]);
    for( int i=n;i>=1;i-- ) a[i]-=a[i-1];
    build(1,1,n);
    while( m-- )
    {
        int opt,l,r,x;
        scanf("%d%d%d",&opt,&l,&r);
        if( opt==1 )
        {
            scanf("%d",&x);
            a[l]+=x;
            if( r!=n ) a[r+1]-=x;
            update(1,1,n,l);update(1,1,n,r+1);
        }
        if( opt==2 ) printf("%d\n",query(1,1,n,l+1,r,1) );
        if( opt==3 )
        printf("%d\n",__gcd( get_sum(1,1,n,1,l),query(1,1,n,l+1,r,0) ) );
    }
}