hdu1166

链接：点击打开链接

题意：第一行一个整数T，表示有T组数据.

每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50).

接下来每行有一条命令,命令有4种形式:

(1)Add(i,j) i和j为正整数,表示第i个营地增加j个(j不超过30)

(2)Sub(i,j) i和j为正整数,表示第i个营地减少j个人(j不超过30)

(3)Query(i,j),i和j为正整数,i<=j,表示询问第i到第j个营地的总人数

(4)End表示结束,这条命令在每组数据最后出现

对第i组数据,首先输出“Case i:”和回车,

对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内

线段树单点更新，求和

代码1：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>
using namespace std;
int node[150000];
void update(int pos,int l,int r,int id,int x){
int mid;
if(l==r){                                 //建数和修改写到一起减少代码长度
node[pos]+=x;
return;
}
else{
mid=(l+r)/2;
if(id<=mid)
update(2*pos,l,mid,id,x);
else
update(2*pos+1,mid+1,r,id,x);
}
node[pos]=node[2*pos]+node[2*pos+1];      //回溯时加上左区间和右区间的值
}
int question(int pos,int l,int r,int ll,int rr){
int mid,sum=0;
if(ll<=l&&rr>=r)return node[pos];
else{
mid=(l+r)/2;
if(ll<=mid)sum+=question(2*pos,l,mid,ll,rr);
if(rr>mid)sum+=question(2*pos+1,mid+1,r,ll,rr);
return sum;
}
}                                 		 //以上为线段树模板，因人而异，也可以写成结构体那种形式
int main(){
int i,k,t,n,x,y;
int a[50005];
char str[100];
scanf("%d",&t);
for(k=1;k<=t;k++){
memset(node,0,sizeof(node));
scanf("%d",&n);
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
update(1,1,n,i,a[i]);
}
for(i=1;i<=100;i++)
cout<<node[i]<<" ";
cout<<endl;
printf("Case %d:\n",k);
while(1){
cin>>str;
if(strcmp(str,"End")==0)
break;
if(strcmp(str,"Add")==0){
scanf("%d%d",&x,&y);
update(1,1,n,x,y);
}
if(strcmp(str,"Sub")==0){
scanf("%d%d",&x,&y);
update(1,1,n,x,-y);
}
if(strcmp(str,"Query")==0){
scanf("%d%d",&x,&y);
printf("%d\n",question(1,1,n,x,y));
}
}
}
return 0;
}

代码2：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int SIZE=50005;
int segtree[SIZE<<2];
void build(int l,int r,int rt){
int m;
if(l==r){
scanf("%d",&segtree[rt]);
return;
}
m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];
}
void update(int p,int add,int l,int r,int rt){
int m;
if(l==r){
segtree[rt]+=add;
return;
}
m=(l+r)>>1;
if(p<=m)
update(p,add,l,m,rt<<1);
else
update(p,add,m+1,r,rt<<1|1);
segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];
}
int query(int L,int R,int l,int r,int rt){
int m,ans;
if(L<=l&&r<=R)
return segtree[rt];
m=(l+r)>>1;
ans=0;
if(L<=m)
ans+=query(L,R,l,m,rt<<1);
if(R>m)
ans+=query(L,R,m+1,r,rt<<1|1);
return ans;
}
int main(){
int t,n,i,j,k,a,b;
char str[100];
scanf("%d",&t);
for(k=1;k<=t;k++){
scanf("%d",&n);
build(1,n,1);
printf("Case %d:\n",k);
while(scanf("%s",str)&&strcmp(str,"End")!=0){
scanf("%d%d",&a,&b);
if(strcmp(str,"Query")==0)
printf("%d\n",query(a,b,1,n,1));
else if(strcmp(str,"Add")==0)
update(a,b,1,n,1);
else
update(a,-b,1,n,1);
}
}
return 0;
}
//强烈推荐看这个讲的线段树，代码特别清晰，适合作为模板
//http://blog.csdn.net/metalseed/article/details/8039326