如果给定一个数组A1,A2,A3…An。求从第i个数到第j个数的和(包含第i个和第j个且1<=i<=j<=n)
很简单直接累加,用sum[i]表A1+A2+…+Ai的和。sum[j]-sum[i-1]就是所求,时间复杂度为O(n)。
#include<cstdio>
const int N = 1000;
int sum[N]; //sum[i]表示A1+A2+...+Ai的和
int A[N]; //A[i]表示数列的第i个数
int main(void)
{
int i, j;
int n;
scanf("%d", &n);
for(i = 1; i <= n; i++) {scanf("%d", &A[i]); sum[i] = sum[i-1]+A[i];}
while(~scanf("%d%d", &i, &j))//询问从第i个数到第j个数的和
printf("%d\n", sum[j] - sum[i-1]);
return 0;
}
如果给定一个数组A1,A2,A3…An。可能的操作为:
- 第i个数加d
- 计算从第i个数到第j个数的和(包含第i个和第j个且1<=i<=j<=n)
类似的题目:hdu1166
如果还用到上面的方法,代码如下:
#include<cstdio>
const int N = 1000;
int sum[N]; //sum[i]表示A1+A2+...+Ai的和
int A[N]; //A[i]表示数列的第i个数
int n;
void update(int i, int d)
{
for(; i <= n; i++) sum[i] += d;
}
int main(void)
{
int i, j;
scanf("%d", &n);
for(i = 1; i <= n; i++) {scanf("%d", &A[i]); sum[i] = sum[i-1]+A[i];}
int m, x, y;
//m表示第m个操作
while(~scanf("%d%d%d", &m, &x, &y))
{
if(m == 1) update(x, y);//m==1时让第x个数加y
else printf("%d\n", sum[y] - sum[x-1]);//m==2时询问从第x个数到第y个数的和
}
return 0;
}
如果有m次操作时间复杂度就为O(m*n),更改第i个数时,sum[i]后面的都要跟新,这样就显得太耗时了。有更好的方法吗?
sum[i]表示A1+A2+…+Ai的和,暂且称为sum[i]管辖的范围为1~i,虽然查询很快,但是当第i个数更改时sum[i],sum[i+1]…sum[n]都要更改。sum数组管辖的范围越大更新的可能越大。有没有方法缩小sum数组管辖的范围。
树状数组就很巧妙的解决了这个问题。
说到树状数组就少不了下面这张图
定义An表示任意n位二进制数,Bn表示各个位全是0的n位二进制数,Cn表示各个位全是1的n位二进制数
树状数组使用二进制数解决了这个问题。An1Bm(二进制数)所管辖的范围为2^m个数。例如当那个数是8(1000二进制)时它管辖的范围为2^3(包含A1..A8),当那个数是6(110二进制)时它管辖的范围为2^1(包含A5,A6),当那个数是5(101二进制)时它管辖的范围为2^0(包含A5)。
假设x的原码为An1Bm
那么-x的反码为A’n0Cm (A’n为An的反码)
那么-x的补码为A’n1Bm
(-x)&x=1Bm=2^m
所以第x个数管辖的范围为(-x)&x。
比如x的原码为00001010
-x的反码为11110101
-x的补码为11110110
11111010&00001010=10=2^1
c[i]表示i管辖的范围内的数和。被管辖的数改变了,管辖的数也要改变。被管辖的数是x时,管辖的数则为x+(-x)&x。(这里说的管辖是直接管辖,所以跟新时管辖点跟新了,管辖点的管辖点也要跟新…)
求第x个数到第y个数的和,就是前y项和减前x-1项和。
求前n项和等于前m(n-第n项管辖的范围)项和加c[n],前m项和只要用同样的方法就可以了。
#include<cstdio>
const int MAXN = 50001;
int c[MAXN];
int n;
//第x个数管辖的范围
int lowbit(int x)
{
return (-x)&x;
}
//求前end项和
int Sum(int end)
{
int sum = 0;
while(end > 0)
{
sum += c[end];
end -= lowbit(end);
}
return sum;
}
//第pos个元素加num
void update(int pos, int num)
{
while(pos <= n)
{
c[pos] += num;
pos += lowbit(pos);
}
}
int main(void)
{
int m, x, y;
int i;
scanf("%d", &n);
int val;
for(i = 1; i <= n; i++)
{
scanf("%d", &val);update(i, val);
}
while(~scanf("%d%d%d", &m, &x, &y))
{
if(m == 2)
{
printf("%d\n", Sum(y)-Sum(x-1));
}
else
{
update(x, y);
}
}
return 0;
}