您的位置:首页 > 其它

树状数组模板

2017-08-17 20:02 148 查看
用树状数组解决问题比线段树(优点)空间复杂度低,并且代码简洁,(缺点)只能求某段和,不能求最值!

#include<cstdio>
#include<cstring>
#define lowerBit(x) (x&(-x))
int num[10000];
int n;
void UpDate(int x,int y) //把第x个数加y
{
while (x <= n)
{
num[x] += y;
x = x+lowerBit(x);
}
}
int Query(int x) //求从1到x的和
{
int sum = 0;
while (x > 0)
{
sum += num[x];
x = x-lowerBit(x); //x&(x-1)
}
return sum;
}
int main()
{
scanf ("%d",&n);
memset(num,0,sizeof(num));
for (int i = 1 ; i <= n ; i++)
{
int t;
scanf ("%d",&t);
UpDate(i,t);
}

int m;
scanf ("%d",&m);
while (m--)
{
int x,y;
scanf ("%d %d",&x,&y);
printf ("%d\n",Query(y)-Query(x-1)); //求解某一段区间的和。
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: