题意:如题
线段树
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int maxn = 1e5+10; ll arr[maxn]; char str[maxn]; /// Pushup 函数更新节点信息,这里是求和 struct SegTree { ll sum[maxn<<2],cnt[maxn<<2],lz[maxn<<2]; void init() /// 初始化 { memset(sum,0,sizeof(sum)); memset(cnt,0,sizeof(cnt)); memset(lz,0,sizeof(lz)); } void push_up(int rt) /// 向上更新(result) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; cnt[rt]=cnt[rt<<1]+cnt[rt<<1|1]; } void push_down(int rt) /// 更下更新(laze) { if(!lz[rt])return; sum[rt<<1]+=cnt[rt<<1]*lz[rt]; sum[rt<<1|1]+=cnt[rt<<1|1]*lz[rt]; lz[rt<<1]+=lz[rt]; lz[rt<<1|1]+=lz[rt]; lz[rt]=0; } void build(ll arr[],int l,int r,int rt) /// 数组建立线段树 { if(l == r) { sum[rt]=arr[l]; cnt[rt] ++; return; } push_down(rt); int mid = (l + r) >> 1; build(arr,l,mid,rt<<1); build(arr,mid+1,r,rt<<1|1); push_up(rt); } void activate(int pos,ll val,int l,int r,int rt) /// 单点建立线段树 { if(l==r) { sum[rt]+=val; cnt[rt]++; return; } push_down(rt); int mid=(l + r) >> 1; if(pos<=mid) activate(pos,val,l,mid,rt<<1); else activate(pos,val,mid+1,r,rt<<1|1); push_up(rt); } void update(int pos,ll val,int l,int r,int rt) /// 单点更新 { if(l == r) { sum[rt] += val; lz[rt] += val; return; } push_down(rt); int mid = (l + r) >> 1; if(pos <= mid) update(pos,val,l,mid,rt<<1); else update(pos,val,mid+1,r,rt<<1|1); push_up(rt); } void update(int ql,int qr,ll val,int l,int r,int rt) /// 区间更新 { if(ql==l && qr==r) { sum[rt]+=cnt[rt]*val; lz[rt]+=val; return; } push_down(rt); int mid=(l + r) >> 1; if(qr<=mid) update(ql,qr,val,l,mid,rt<<1); else if(ql>mid) update(ql,qr,val,mid+1,r,rt<<1|1); else { update(ql,mid,val,l,mid,rt<<1); update(mid+1,qr,val,mid+1,r,rt<<1|1); } push_up(rt); } ll query(int ql,int qr,int l,int r,int rt) /// 区间查询 { if(ql==l && qr==r)return sum[rt]; push_down(rt); int mid=(l + r) >> 1; if(qr<=mid)return query(ql,qr,l,mid,rt<<1); if(ql>mid)return query(ql,qr,mid+1,r,rt<<1|1); return query(ql,mid,l,mid,rt<<1)+query(mid+1,qr,mid+1,r,rt<<1|1); } }seg; int main() { int caset,t = 0;scanf("%d",&caset); while(caset--) { int n,l,r,pos; ll x; seg.init(); scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%lld",&arr[i]); seg.build(arr,1,n,1); printf("Case %d:\n",++t); while(scanf("%s",str) && str[0] != 'E') { if(str[0] == 'A') { scanf("%d%lld",&pos,&x); seg.update(pos,x,1,n,1); } else if(str[0] == 'S') { scanf("%d%lld",&pos,&x); seg.update(pos,-x,1,n,1); } else if(str[0] == 'Q') { scanf("%d%d",&l,&r); printf("%lld\n",seg.query(l,r,1,n,1)); } } } return 0; }
我们知道单点修改+区间查询可以用树状数组来做
树状数组代码
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int maxn = 1e5+10; ll arr[maxn]; char str[maxn]; struct BIT { ll c[maxn]; ll lowbit(ll x) { return x&-x; } void init() { memset(c,0,sizeof(c)); } void Update(int pos,ll x,int n) { for(int i=pos;i<=n;i += lowbit(i)) c[i] += x; } ll Query(int pos) { ll ans = 0; for(int i=pos;i;i-=lowbit(i)) ans += c[i]; return ans; } }bit; int main() { int caset,t = 0;scanf("%d",&caset); while(caset--) { printf("Case %d:\n",++t); bit.init(); int n,l,r,pos; ll x; scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%lld",&arr[i]),bit.Update(i,arr[i],n); while(scanf("%s",str) && str[0] != 'E') { if(str[0] == 'A') { scanf("%d%lld",&pos,&x); bit.Update(pos,x,n); } else if(str[0] == 'S') { scanf("%d%lld",&pos,&x); bit.Update(pos,-x,n); } else if(str[0] == 'Q') { scanf("%d%d",&l,&r); printf("%lld\n",bit.Query(r) - bit.Query(l-1)); } } } return 0; }