题目要求区间修改和单点查询,不难想到线段树,区间修改为区间从左到右依次加一个等差数列,则可以用线段树+差分解决,使用差分数组则可以实现区间加同一个数来实现区间加等差数列
eg:初始数组:0 0 0 0 0
等差数列:首项1,公差2,从下标1加到4
操作之后的序列:1 3 5 7 0
差分数组:1 2 2 2 -7
所以只需要将l位置加首项,l+1到r区间加等差,r+1减去末尾项即可实现区间加等差数列操作
对于单点查询只需要查询前缀和即可
注意避免数组越界问题
ac代码:
#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
#define pb push_back
#define int long long
#define Mirai ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
typedef pair<int,int> pii;
const int N=1e5+10;
struct Tree
{
int l,r;
int sum,lazy;
}tr[N<<2];
int w[N];
int n,m,op,l,r,k,d;
void change(int u,int lazy)
{
tr[u].sum+=(tr[u].r-tr[u].l+1)*lazy;
tr[u].lazy+=lazy;
}
void pushdown(int u)
{
if(tr[u].lazy)
{
change(u<<1,tr[u].lazy);
change(u<<1|1,tr[u].lazy);
tr[u].lazy=0;
}
}
void pushup(int u)
{
tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void build(int u,int l,int r)
{
if(l==r)tr[u]={l,r,w[l]-w[l-1],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 k)
{
if(tr[u].l>=l&&tr[u].r<=r)
{
change(u,k);
return ;
}
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
if(l<=mid)modify(u<<1,l,r,k);
if(r>mid)modify(u<<1|1,l,r,k);
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 res=0;
if(l<=mid)res+=query(u<<1,l,r);
if(r>mid)res+=query(u<<1|1,l,r);
return res;
}
void solve()
{
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>w[i];
build(1,1,n);
while(m--)
{
cin>>op;
if(op==1)
{
cin>>l>>r>>k>>d;
modify(1,l,l,k);
if(l+1<=r)modify(1,l+1,r,d);
if(r<n)modify(1,r+1,r+1,-1*(k+(r-l)*d));
}
else
{
cin>>l;
cout<<query(1,1,l)<<endl;
}
}
}
signed main()
{
Mirai;
int T=1;
//cin>>T;
while(T--)
{
solve();
}
}