题目描述
维护一个数列 (a_i)，支持两种操作：
1 l r K D：给出一个长度等于 (r-l+1) 的等差数列，首项为 (K)，公差为 (D)，并将它对应加到 ([l,r]) 范围中的每一个数上。即：令 (a_l=a_l+K,a_{l+1}=a_{l+1}+K+Dldots a_r=a_r+K+(r-l) imes)。
2 p：询问序列的第 (p) 个数的值 (a_p)。
输入格式
第一行两个整数数 (n,m) 表示数列长度和操作个数。
第二行 (n) 个整数，第 (i) 个数表示 (a_i)。
接下来的 (m) 行，每行先输入一个整数 (opt)。
若 (opt=1) 则再输入四个整数 (l r K D)；
若 (opt=2) 则再输入一个整数 (p)。
输出格式
对于每个询问，一行一个整数表示答案。
差分+线段树
就没什么好说的吧
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=2e5+10;
#define int long long
int a[N],n,m;
struct Seg{
int l,r,sum,add;
#define l(x) tree[x].l
#define r(x) tree[x].r
#define sum(x) tree[x].sum
#define add(x) tree[x].add
#define len(x) (r(x)-l(x)+1)
}tree[N<<2];
#define ls p<<1
#define rs p<<1|1
#define mid ((l(p)+r(p))>>1)
void build(int p,int l,int r){
l(p)=l,r(p)=r;
if(l==r){
sum(p)=a[l];
return;
}
build(ls,l,mid);
build(rs,mid+1,r);
sum(p)=sum(ls)+sum(rs);
}
inline void pushdown(int p){
sum(ls)+=add(p)*len(ls);
sum(rs)+=add(p)*len(rs);
add(ls)+=add(p);
add(rs)+=add(p);
add(p)=0;
}
void update(int p,int l,int r,int d){
if(l<=l(p)&&r(p)<=r){
sum(p)+=len(p)*d;
add(p)+=d;
return ;
}
if(add(p))pushdown(p);
if(l<=mid)update(ls,l,r,d);
if(r>mid)update(rs,l,r,d);
sum(p)=sum(ls)+sum(rs);
}
int query(int p,int l,int r){
if(l<=l(p)&&r(p)<=r)return sum(p);
if(add(p))pushdown(p);
int ans=0;
if(l<=mid)ans+=query(ls,l,r);
if(r>mid)ans+=query(rs,l,r);
sum(p)=sum(ls)+sum(rs);
return ans;
}
signed main(){
cin>>n>>m;
for(int i=1,x=0,y=0;i<=n;i++){
scanf("%lld",&x);
a[i]=x-y;
y=x;
}
build(1,1,n);
int opt,l,r,k,d;
while(m--){
scanf("%lld",&opt);
if(opt==1){
scanf("%lld%lld%lld%lld",&l,&r,&k,&d);
update(1,l,l,k);
if(l!=r)update(1,l+1,r,d);
if(r<n)update(1,r+1,r+1,-k-(r-l)*d);
}else{
scanf("%lld",&k);
printf("%lld
",query(1,1,k));
}
}
}