牛客算法周周练8 C ( 线段树 + 裴蜀定理 )

题目链接

解题报告：
我们用一颗线段树维护差分数组，同时也维护 区间和 ，区间最大值，区间gcd 。
对于操作1：只需让 {l}+x，{r+1}-=x。
对于操作2：本身维护的就是差分数组，直接去max。
对于操作3：gcd({l}区间和，{l+1,r} 区间gcd)。

gcd（a,b）= gcd（a-b,b） =>> gcd（a,b,c,d）=gcd（a,b-a,c-b,d-c）；

#define first f
#define second s
#define ll long long
#define mp make_pair
#define pb push_back
#define pf push_front
#define lb lower_bound
#define ub upper_bound
#define sl(p) strlen(p)
#define SZ(p) p.size()
#define pii pair<int,int>
#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))

using namespace std;
const int MOD=1e9+7;
const int maxn=1e5+5;
const int inf=0x3f3f3f3f;
int n,m;
int Max[maxn<<2],sum[maxn<<2],Gcd[maxn<<2],a[maxn];

int my_max(int x,int y)
{
    return abs(x)>abs(y)?abs(x):abs(y);
}
int query3(int L,int R,int l,int r,int rt)/** 最大值 **/
{
    if(L<=l&&R>=r) return Max[rt];
    int mid=(l+r)>>1;
    int ans=0;
    if(L<=mid) ans=my_max(ans,query3(L,R,l,mid,rt<<1));
    if(R>mid) ans=my_max(ans,query3(L,R,mid+1,r,rt<<1|1));
    return ans;
}
int query2(int L,int R,int l,int r,int rt) /** 区间gcd **/
{
    if(L<=l&&R>=r) return Gcd[rt];
    int mid=(l+r)>>1;
    int ans=0;
    if(L<=mid) ans=__gcd(ans,query2(L,R,l,mid,rt<<1));
    if(R>mid) ans=__gcd(ans,query2(L,R,mid+1,r,rt<<1|1));
    return ans;
}
int query1(int L,int R,int l,int r,int rt)/** a[i] **/
{
    if(L<=l&&R>=r) return sum[rt];
    int mid=(l+r)>>1;
    int ans=0;
    if(L<=mid) ans+=query1(L,R,l,mid,rt<<1);
    if(R>mid) ans+=query1(L,R,mid+1,r,rt<<1|1);
    return ans;
}
void pushup(int rt)
{
    Max[rt]=my_max(Max[rt<<1],Max[rt<<1|1]);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    Gcd[rt]=__gcd(Gcd[rt<<1],Gcd[rt<<1|1]);
}
void build(int l,int r,int rt)
{
    if(l==r){
        sum[rt]=Gcd[rt]=Max[rt]=a[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
void update(int pos,int l,int r,int rt,int val)
{
    if(l==r){
        Max[rt]=sum[rt]=Gcd[rt]=Max[rt]+val;
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) update(pos,l,mid,rt<<1,val);
    else update(pos,mid+1,r,rt<<1|1,val);
    pushup(rt);
}
int main()
{
    int l,r,opt,x;
    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,n,1);
    for(int i=1;i<=m;i++){
        scanf("%d",&opt);
        if(opt==1){
            scanf("%d%d%d",&l,&r,&x);
            update(l,1,n,1,x);
            if(r<n) update(r+1,1,n,1,-x);
        }
        else if(opt==2){
            scanf("%d%d",&l,&r);
            if(l==r) printf("0\n");
            else printf("%d\n",abs(query3(l+1,r,1,n,1)));
        }
        else{
            scanf("%d%d",&l,&r);
            if(l==r) printf("%d\n",query1(1,l,1,n,1));
            else printf("%d\n",abs(__gcd(query1(1,l,1,n,1),query2(l+1,r,1,n,1))));
        }
    }
    return 0;
}