题目传送门
一个既可以用树状数组又可以用线段树写的题。(不过树状数组好像耗时更多,但是写起来比线段树香)
题目就是给一个原数组,要么对一个区间内一起增加或者减少一个数值,要么就询问该段区间的和。
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+5;
const int mod=1e9+7;
const int INF=0x7fffffff;
const ll LLINF=0x7fffffffffffffff;
const double EPS=1e-10;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define pii pair<int,int>
#define pll pair<ll,ll>
#define debug cout<<"debug"<<endl;
//#define int long long
int a[N];
struct tree
{
int l,r,pre,add;
}t[N<<2];
void build(int p,int l,int r)
{
t[p].l=l;t[p].r=r;
if(l==r)
{
t[p].pre=a[l];
return ;
}
int mid=t[p].l+t[p].r>>1;
build(p<<1,l,mid);
build(p<<1|1,mid+1,r);
t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
void spread(int p)
{
if(t[p].add)
{
t[p<<1].pre+=t[p].add*(t[p<<1].r-t[p<<1].l+1);
t[p<<1|1].pre+=t[p].add*(t[p<<1|1].r-t[p<<1|1].l+1);
t[p<<1].add+=t[p].add;
t[p<<1|1].add+=t[p].add;
t[p].add=0;
}
}
void change(int p,int x,int y,int z)
{
if(x<=t[p].l&&t[p].r<=y)
{
t[p].pre+=z*(t[p].r-t[p].l+1);
t[p].add+=z;
return ;
}
spread(p);
int mid=t[p].r+t[p].l>>1;
if(x<=mid)
change(p<<1,x,y,z);
if(y>mid)
change(p<<1|1,x,y,z);
t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
int ask(int p,int x,int y)
{
if(x<=t[p].l&&t[p].r<=y)
{
return t[p].pre;
}
spread(p);
int mid=t[p].l+t[p].r>>1;
int ans=0;
if(x<=mid)
ans+=ask(p<<1,x,y);
if(y>mid)
ans+=ask(p<<1|1,x,y);
return ans;
}
signed main()
{
IOS;
//freopen("","r",stdin);
//freopen("","w",stdout);
int T;
cin>>T;
int cnt=0;
while(T--)
{
memset(t,0,sizeof t);
cnt++;
cout<<"Case "<<cnt<<":"<<endl;
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
build(1,1,n);
string s;
while(cin>>s)
{
if(s=="End")
break;
if(s=="Add")
{
int i,j;
cin>>i>>j;
change(1,i,i,j);
}
if(s=="Sub")
{
int i,j;
cin>>i>>j;
change(1,i,i,-j);
}
if(s=="Query")
{
int i,j;
cin>>i>>j;
int res=ask(1,i,j);
cout<<res<<endl;
}
}
}
}