https://vjudge.net/problem/HDU-1166
单点更新,区间查询。
#include <bits/stdc++.h>
/* 分治的思想,维护一个满二叉树。
每个点 保存两个点,左面的和右面的
*/
using namespace std;
const int maxn=50007;
struct Node
{ int l,r;
int sum;
}node[maxn*4];
int a[maxn];
void build(int num,int lef,int rig){
node[num].l=lef;
node[num].r=rig;
if(lef==rig)
{ //cout<<num<<endl;
node[num].sum=a[lef];//更新到单点
return ;
}
int mid=(lef+rig)/2;
build(num<<1,lef,mid);//更新左孩子
build(num<<1|1,mid+1,rig);//更新右孩子
node[num].sum=node[num<<1].sum+node[num<<1|1].sum;
}
void update(int num,int pos,int v){
if(node[num].l==node[num].r)
{//node[num].value+=v;//完全包围的区间,全部加入
node[num].sum=v;
return ;
}
int mid=(node[num].l+node[num].r)/2;
if(pos<=mid)
update(num*2,pos,v);//横跨左孩子
if(pos>mid)
update(num*2+1,pos,v);
node[num].sum=node[num*2].sum+node[num*2+1].sum;
}
int query(int num,int le,int re){
if(le<=node[num].l&&re>=node[num].r){
return node[num].sum;//该区间包含查询区间,看错了
}//区间内进行更新。最简单的形式,最后都会递归到这一层
int mid=node[num].l+node[num].r;
int all=0;
if(le<=(mid/2))
all+=query(num<<1,le,re);
if(re>(mid/2))//横跨右边
all+=query(num<<1|1,le,re);//右边
return all;
}
int main()
{ int t;
int m;
string s;
int a1,b1;
scanf("%d",&t);
for(int tt=1;tt<=t;tt++){
scanf("%d",&m);
memset(a,0,sizeof(a));
for(int i=1;i<=m;i++)
scanf("%d",&a[i]);
build(1,1,m);
//cout<<"!!!!!"<<endl;
printf("Case %d:\n",tt);
while(cin>>s){
//cout<<s<<endl;
if(s=="End") break;
else if(s=="Query"){
scanf("%d%d",&a1,&b1);
if(a1>b1) swap(a1,b1);
printf("%d\n",query(1,a1,b1));
}
else if(s=="Add")
{ scanf("%d%d",&a1,&b1);
a[a1]+=b1;
update(1,a1,a[a1]);
}
else if(s=="Sub"){
scanf("%d%d",&a1,&b1);
a[a1]-=b1;
update(1,a1,a[a1]);
}
}
}
return 0;
}