Virtual Judge上的敌兵布阵
工兵营地好比数组,这里要任意取一段工兵营地计算人数,很明显此题可以采用树状数组或线段树求解的。与上篇博客中树状数组不同,此题树状数组的实现采取结构体型,用c++中的类进行封装如下:
本题中还有对树状数组元素的增减查询操作:
数据更新:
void Tree::updata(int x, int y) //对指定军营人数进行变更
{
x++;
while (x <= n)
{
C[x] += y;
x = x+lowbit(x);
}
}
数据求和:
int Tree::Add(int i) //计算指定军营的人数和
{
i++;
int sum = 0;
while (i>0)
{
sum += C[i];
i = i-lowbit(i);
}
return sum;
}