P1438 无聊的数列 线段树+差分

P1438 无聊的数列 线段树+差分

---

思路: 该题无非是在 线段树1 的基础上加了差分数组,那么我们着重需要讨论的就是 如何维护差分数组 。已知当 $opt=1$ 时，我们需要在一段区间 $[l,r]$ 内加上首先是 $k$ 公差为 $d$ 的等比数列。可以通过以下推导进行线段树修改：

• $l=1$ $r=5$ $k=1$ $d=2$
• 在下标 $[1,5]$ 中分别加上 $1$ $3$ $5$ $7$ $9$
• 将其转化为差分数组：$1$ $2$ $2$ $2$ $2$ $-9$
• 通过上式可知我们仅需进行 $3$ 次 $modify$ 操作，分别是 $modify(1,l,l)$ 、$modify(1,l+1,r)$ 、$modify(1,r+1,r+1)$
• 需要注意的细节，如果 $l==r$ 那么无须 $modify(1,l+1,r)$ 如果 $r+1>n$ 那么无须 $modify(1,r+1,r+1)$ 。

---

代码：

/*
 * @author:  Snow
 * @Description:  Algorithm Contest
 * @LastEditTime: 2022-07-26 09:43:44
 */
#include<bits/stdc++.h>
using namespace std;
#define int long long 
// #pragma GCC optimize(3)
#define re register int
typedef pair<int,int>PII;
#define pb emplace_back
#define debug(a) cout<<a<<' ';
#define fer(i,a,b) for(re i=a;i<=b;i++)
#define der(i,a,b) for(re i=a;i>=b;i--)
int n,m;
const int N = 1e5+10;
int w[N];
struct Node{
    int l,r,sum,lazy;
}tr[N*4];
void pushup(int u){
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void pushdown(int u){
    auto &root=tr[u],&left=tr[u<<1],&right=tr[u<<1|1];
    if(tr[u].lazy){
        left.lazy+=root.lazy,left.sum+=(left.r-left.l+1)*root.lazy;
        right.lazy+=root.lazy,right.sum+=(right.r-right.l+1)*root.lazy;
        root.lazy=0;
    }
}
void build(int u,int l,int r){
    if(l==r){
        tr[u]={l,r,w[l],0};
    }
    else{
        tr[u]={l,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 l,int r,int d){
    if(tr[u].l>=l&&tr[u].r<=r){
        tr[u].sum+=(tr[u].r-tr[u].l+1)*d;
        tr[u].lazy+=d;//延迟下放
    }
    else{
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(l<=mid)modify(u<<1,l,r,d);
        if(r>mid)modify(u<<1|1,l,r,d);
        pushup(u);
    }
}
int query(int u,int l,int r){
    if(tr[u].l>=l&&tr[u].r<=r){
        return tr[u].sum;
    }
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    int sum=0;
    if(l<=mid)sum=query(u<<1,l,r);
    if(r>mid)sum+=query(u<<1|1,l,r);
    return sum;
}
signed main(){
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>m;
    fer(i,1,n)cin>>w[i];
    der(i,n,1)w[i]=w[i]-w[i-1];
    build(1,1,n);
    while(m--){
        int op;
        cin>>op;
        if(op==1){
            int l,r,k,d;
            cin>>l>>r>>k>>d;
            modify(1,l,l,k);
            if(l<r)modify(1,l+1,r,d);
            if(r<n)modify(1,r+1,r+1,-(k+(r-l)*d));
        }
        else{
            int p;
            cin>>p;
            cout<<query(1,1,p)<<endl;
        }
    }
    return 0;
}