P3372 【模板】线段树 1
非常正规的板子题了可以说
P3372 【模板】线段树 1
题解
注意几个点就好啦
(1)懒标记 add[ ],数组一定要开的大一点
前缀和 sum[ ],数组一定要开的大一点,还要开long long
(2)build 建树函数:树 k ,边界 l ~ r
(3)Add 加数函数:树 k ,边界 l ~ r ,区间加 p
(4)pushdown 标记下传函数:树 k ,边界 l ~ r ,中间 mid
(5)modify 区间加数函数:树 k ,边界 l ~ r ,询问区间 x ~ y ,区间加数 v
(6)quety 区间查询函数:树 k ,边界 l ~ r ,询问区间 x ~ y
代码
#include<bits/stdc++.h> using namespace std; const int maxn=1e5+10; int n,m,x,y,v,opr; int a[maxn],add[maxn*4]; long long sum[maxn*4]; long long ans; inline int read() { int ans=0; char last=' ',ch=getchar(); while(ch<'0'||ch>'9') last=ch,ch=getchar(); while(ch>='0'&&ch<='9') ans=ans*10+ch-'0',ch=getchar(); if(last=='-') ans=-ans; return ans; } void build(int k,int l,int r) { if(l==r) { sum[k]=a[l]; return ; } int mid=(l+r)>>1; build(k<<1,l,mid); build(k<<1|1,mid+1,r); sum[k]=sum[k<<1]+sum[k<<1|1]; } void Add(int k,int l,int r,int p) { add[k]+=p; sum[k]+=(long long)(r-l+1)*p; } void pushdown(int k,int l,int r,int mid) { if(add[k]==0) return ; Add(k<<1,l,mid,add[k]); Add(k<<1|1,mid+1,r,add[k]); add[k]=0; } void modify(int k,int l,int r,int x,int y,int v) { if(x<=l&&r<=y) return Add(k,l,r,v); int mid=(l+r)>>1; pushdown(k,l,r,mid); if(x<=mid) modify(k<<1,l,mid,x,y,v); if(mid<y) modify(k<<1|1,mid+1,r,x,y,v); sum[k]=sum[k<<1]+sum[k<<1|1]; } long long query(int k,int l,int r,int x,int y) { if(x<=l&&r<=y) return sum[k]; int mid=(l+r)>>1; long long res=0; pushdown(k,l,r,mid); if(x<=mid) res+=query(k<<1,l,mid,x,y); if(mid<y) res+=query(k<<1|1,mid+1,r,x,y); return res; } int main() { n=read();m=read(); for(int i=1;i<=n;i++) a[i]=read(); build(1,1,n); for(int i=1;i<=m;i++) { opr=read();x=read();y=read(); if(opr==1) { v=read(); modify(1,1,n,x,y,v); } if(opr==2) { ans=query(1,1,n,x,y); printf("%lld\n",ans); } } return 0; }