POJ 3468 A Simple Problem with Integers【线段树】-优快云博客

本文介绍了一种基于树状数组实现的区间更新与查询算法，通过定义一系列宏指令简化了段树的操作，并实现了高效的区间加法更新及区间求和查询功能。适用于处理大规模数据集的动态更新和查询需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream> #include<cstdio> using namespace std; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 #define mid (l+r)>>1 typedef long long ll; const int maxn=100000+10; ll sum[maxn<<2],add[maxn<<2]; inline void read(ll &d){ char ch;int op=1; while(ch=getchar(),(ch>57||ch<48)&&ch!=45);ch==45?op=-1,d=0:d=ch-48; while(ch=getchar(),ch>47&&ch<58) d=d*10+ch-48; d*=op; } inline void write(ll d){ if(d==0){puts("0");return;} if(d<0) putchar(45),d=-d; char s[20];s[19]=0; int i=18; while(d){ s[i]=d%10+48; d/=10;i--; } puts(s+i+1); } inline void pushup(int rt){sum[rt]=sum[rt<<1]+sum[rt<<1|1];} inline void pushdown(int rt,int m) { if(!add[rt]) return; add[rt<<1]+=add[rt]; add[rt<<1|1]+=add[rt]; sum[rt<<1]+=add[rt]*(m-(m>>1)); sum[rt<<1|1]+=add[rt]*(m>>1); add[rt]=0; } void build(int l,int r,int rt) { add[rt]=0; if(l==r){read(sum[rt]);return;} int m=mid; build(lson);build(rson); pushup(rt); } void update(int L,int R,ll c,int l,int r,int rt) { if(L<=l&&r<=R) { add[rt]+=c; sum[rt]+=c*(r-l+1); return; } int m=mid; pushdown(rt,r-l+1); if(L<=m) update(L,R,c,lson); if(R>m) update(L,R,c,rson); pushup(rt); } ll query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R) return sum[rt]; pushdown(rt,r-l+1); int m=mid; ll ret=0; if(L<=m) ret+=query(L,R,lson); if(m<R) ret+=query(L,R,rson); return ret; } int main() { int n,q; scanf("%d%d",&n,&q); build(1,n,1); char ch; for(int i=0;i<q;i++) { ll x,y,z; ch=getchar(); read(x); read(y); if(ch=='Q') write(query(x,y,1,n,1)); else read(z),update(x,y,z,1,n,1); } return 0; }