773
京东网上商城
HDU 1166 线段树单点更新
第一行一个整数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 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define N 50005
using namespace std;
int num[N];
struct Tree
{
int l; //左端点
int r; //右端点
int sum; //总数
} tree[N*4]; // 总线段的长度为N，开数组的话一般开到N 的四倍
void build(int root,int l,int r) // root表示根节点 ，他的区间范围【l,r】
{
tree[root].l=l;
tree[root].r=r;
if(tree[root].l==tree[root].r) // 当左右端点相等时就是叶子节点
{
tree[root].sum=num[l]; // 赋除值
return; // 递归出口
}
int mid=(l+r)/2;
build(root<<1,l,mid); // k<<1相等于k*2 即是他的左孩子
build(root<<1|1,mid+1,r); // k<<1|1 相当于k*2+1 ,即是他的右孩子
tree[root].sum = tree[root<<1].sum + tree[root<<1|1].sum; // 父亲的sum = 左孩子的sum + 右孩子的sum
}
void update(int root,int pos,int val) // root是根节点，pos，val表示：我们要跟新在pos点出的值更新为val
{
if(tree[root].l==tree[root].r) // 如果是叶子节点，即是pos对应的位置
{
tree[root].sum=val; // 更新操作
return; // 递归出口
}
int mid=(tree[root].l + tree[root].r)/2;
if(pos<=mid) // 如果pos点是在root对应的左孩子的话，就调用update(k<<1,pos,val); 在左孩子里找
update(root<<1,pos,val);
else
update(root<<1|1,pos,val);
tree[root].sum = tree[root<<1].sum + tree[root<<1|1].sum; // 父亲的sum = 左孩子的sum + 右孩子的sum
}
int query(int root,int L,int R) // root表示根节点，[L,R]表示要查询的区间
{
if(L<=tree[root].l&&R>=tree[root].r) // [L,R]要查询的区间 包含root节点表示的区间 直接返回root节点的sum值
return tree[root].sum;
int mid=(tree[root].l + tree[root].r)/2,ret=0;
if(L<=mid) ret+=query(root<<1,L,R); // 查询root节点的左孩子
if(R>mid) ret+=query(root<<1|1,L,R); // 查询root节点的右孩子
return ret; // 返回
}
int main()
{
int t,n,ca=0;
char s[10];
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(int i=1; i<=n; i++)
scanf("%d",&num[i]);
build(1,1,n);
printf("Case %d:\n",++ca);
while(scanf("%s",s),s[0]!='E')
{
int a,b;
scanf("%d%d",&a,&b);
if(s[0]=='Q')
{
if(a>b)
swap(a,b);
printf("%d\n",query(1,a,b));
}
if(s[0]=='A')
num[a]+=b,update(1,a,num[a]);
if(s[0]=='S')
num[a]-=b,update(1,a,num[a]);
}
}
return 0;
}
最后更新：2017-04-03 18:51:56