【完全版】线段树初学--HDU 1166 敌兵布阵

链接:click here~~ 

题意:

第一行一个整数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条命令

Output

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

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

Sample Input

1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End

Sample Output

Case 1: 6 33 59

【解题思路】单点更新:最最基础的线段树,只更新叶子节点,然后把信息用PushUP(int r)这个函数更新上来

代码:


#include#include#include#include#include#include#include#include#include#includeusing namespace std;int dir[4][2]= {{1,0},{-1,0},{0,1},{0,-1}};#define mem(a) memset(a,0,sizeof(a))#define LL long long#define Max(a,b) a>b?a:b;#define Min(a,b) a>1;

build(lchild);

build(rchild);

Pushup(rt);

}

void update(int p,int add,int l,int r,int rt)  // 单点更新

{

if(l==r)

{

tree[rt]+=add;

return ;

}

int m=(l+r)>>1;

if(p<=m) update(p,add,lchild);

else update(p,add,rchild);

Pushup(rt);

}

int query(int L,int R,int l,int r,int rt)      //区间查询

{

if(L<=l&&r<=R)

{

return tree[rt];

}

int m=(l+r)>>1;

int sum=0;

if(L<=m) sum+=query(L,R,lchild);

if(R>m) sum+=query(L,R,rchild);

return sum;

}

int scan()

{

int res = 0, flag = 0;

char ch;

if((ch = getchar()) == '-') flag = 1;

else if(ch >= '0' && ch <= '9') res = ch - '0';

while((ch = getchar()) >= '0' && ch <= '9')

res = res * 10 + (ch - '0');

return flag ? -res : res;

}

int main()

{

int t,n,m,i,j,tot=1;

scanf("%d",&t);

while(t--)

{

printf("Case %d:\n",tot++);

n=scan();

build(1,n,1);

char op[10];

while(scanf("%s",op))

{

if(op[0]=='E') break;

int a,b;

a=scan();

b=scan();

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;

}

你可能感兴趣的:(【完全版】线段树初学--HDU 1166 敌兵布阵)