HDU 1166 敌兵布阵 [线段树-单点更新]

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define MAXSIZE 300000
#define BUF_SIZE 50
#define L 0
#define R 1
#define SUM 2

long seg[MAXSIZE][3];

void build(long r,long x1,long x2)
{
seg[r][L]=x1;
seg[r][R]=x2;
seg[r][SUM]=0;
if(x1==x2)
return ;
else
{
long mid=(x1+x2)>>1;
build(r<<1,x1,mid);
build((r<<1)+1,mid+1,x2);
}
}

void add(long id,long n)
{
long x1,x2,mid,r;

r=1;
x1=seg[r][L];
x2=seg[r][R];
mid=(x1+x2)>>1;

while(x2!=x1)
{
seg[r][SUM]+=n;
if(id<=mid)
r<<=1;//left
else
r=(r<<1)+1;//right
x1=seg[r][L];
x2=seg[r][R];
mid=(x1+x2)>>1;
}
seg[r][SUM]+=n;
}

long query(long r,long x1,long x2)
{
long left=seg[r][L],right=seg[r][R],mid=(left+right)>>1;

if(x1==left&&x2==right)
return seg[r][SUM];
else if(x2<=mid)
return query(r<<1,x1,x2);
else if(x1>mid)
return query((r<<1)+1,x1,x2);
else
return query(r<<1,x1,mid)+query((r<<1)+1,mid+1,x2);
}

int main()
{
int ncase,ccase=0;
long N,i,j;
char buf[BUF_SIZE];

scanf("%d",&ncase);
while(ncase--)
{
printf("Case %d:\n",++ccase);

scanf("%ld",&N);
build(1,1,N);
for(i=1; i<=N; i++)
{
scanf("%ld",&j);
add(i,j);
}
while(scanf("%s",buf),*buf!='E')
{
scanf("%ld%ld",&i,&j);
if(*buf=='A')
add(i,j);
else if(*buf=='S')
add(i,-j);
else
printf("%ld\n",query(1,i,j));
}
}
return 0;
}

#include <cstdio>

using namespace std;

const int maxn = 55555;
int sum[maxn<<2];

void PushUP(int rt)
{
sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

void build(int l,int r,int rt)
{
if (l == r)
{
scanf("%d",&sum[rt]);
return ;
}
int m = (l + r) >> 1;
build(l , m , rt << 1);
build(m + 1 , r , rt << 1 | 1);
PushUP(rt);
}

void update(int p,int add,int l,int r,int rt)
{
if (l == r)
{
sum[rt] += add;
return ;
}
int m = (l + r) >> 1;
if (p <= m)
update(p , add , l , m , rt << 1);
else
update(p , add , m + 1 , r , rt << 1 | 1);
PushUP(rt);
}

int query(int L,int R,int l,int r,int rt)
{
if (L <= l && r <= R)
{
return sum[rt];
}
int m = (l + r) >> 1;
int ret = 0;
if (L <= m)
ret += query(L , R , l , m , rt << 1);
if (R > m)
ret += query(L , R , m + 1 , r , rt << 1 | 1);
return ret;
}

int main()
{
int T , n;
scanf("%d",&T);
for (int cas = 1 ; cas <= T ; cas ++)
{
printf("Case %d:\n",cas);
scanf("%d",&n);
build(1 , n , 1);
char op[10];
while (scanf("%s",op))
{
if (op[0] == 'E')
break;
int a , b;
scanf("%d%d",&a,&b);
if (op[0] == 'Q')
printf("%d\n",query(a , b , 1 , n , 1));
else if (op[0] == 'S')
update(a , -b , 1 , n , 1);
else
update(a , b , 1 , n , 1);
}
}
return 0;
}