线段树第一题。
hdu1166 敌兵布阵
题目大意:给定 N 个工兵营地,编号为 1--N,每个营地有相应的人数。现在对连续的工兵营地进行操作:
(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 表示结束,这条命令在每组数据最后出现;
#include <iostream>
#include <cstdio>
#include <cstring>
#define Max 50005
#define LL(x) ((x)<<1)
#define RR(x) ((x)<<1|1)
#define FF(i,n) for(int i = 0 ; i < n ; i ++)
using namespace std;
struct Node
{
int left,right,num;
int calmid()
{
return (left+right)/2;
}
}tt[Max<<2];
int num[Max];
//当left==right时,元线段。叶子节点
int build(int left,int right,int idx)
{
tt[idx].left = left;
tt[idx].right = right;
if(left == right)
{
return tt[idx].num = num[left];
}
int mid = (left + right)/2;
return tt[idx].num = build(left,mid,LL(idx)) + build(mid+1,right,RR(idx));
}
void update(int id,int x,int idx)
//从 idx 号节点往下更新,第id是元线段
{
tt[idx].num += x;
//num + = x (每次加 X)
if(tt[idx].left == tt[idx].right)
//递归进行,直到查找到元线段
{
return ;
}
int mid = tt[idx].calmid();
if(id <= mid)
{
update(id,x,LL(idx));
//
/2
}
else
update(id,x,RR(idx));
//
/2+1
}
}
int query(int left,int right,int idx)
{
if(left == tt[idx].left && right == tt[idx].right)
{
return tt[idx].num;
}
int mid = tt[idx].calmid();
if(right <= mid)
{
return query(left,right,LL(idx));
}
else if(mid < left)
{
return query(left,right,RR(idx));
}
else
{
return query(left,mid,LL(idx)) + query(mid+1,right,RR(idx));
}
}
int main()
{
update(a,b,1);
break;
update(a,-b,1);
break;
}
}