自用模板,原理网上一大堆,不过建议自己跟着题目敲几遍,有部分所谓题解代码是错的。
<树状数组区间更新与查询>
#include<bits/stdc++.h> using namespace std; const int maxn=50005; int n,t,s; //n数组长度,t区间修改次数,s查询次数 int a[maxn]= {0}; int sum1[maxn]= {0}; int sum2[maxn]= {0}; int lowbit(int x) { return x&(-x); } int update(int i,int k) { int rec=i; while(i<=n) { sum1[i]+=k; sum2[i]+=k*rec; i+=lowbit(i); } } int getsum(int i) { int res=0,cnt=i; while(i>0) { res+=(cnt+1)*sum1[i]-sum2[i]; i-=lowbit(i); } return res; } int main() { cin>>n; for(int i=1; i<=n; i++) { cin>>a[i]; update(i,a[i]-a[i-1]); } cin>>t; for(int i=1; i<=t; i++) { int l,r,k; //[l,r]区间 ,k为修改的价值 cin>>l>>r>>k; update(l,k); update(r+1,-k); } /*for(int i=0;i<=n+1;i++){ printf("%d\t%d\t%d\t%d\n",a[i],sum1[i],sum2[i],getsum(i)); }*/ cin>>s; for(int i=1; i<=s; i++) { int l,r; cin>>l>>r; cout<<getsum(r)-getsum(l-1)<<endl; //如果查询较多的话可以这个结果也用一个数组存起来 } }
0.0分
0 人评分