单点更新,区间求和
#include <iostream>
#include <cstdio>
using namespacestd;
const int maxn =1e6 + 1;
int a[maxn],c[maxn];
int maxnum = 1e6;
int lowbit(int x) {return x & -x;}
int sum(int x)
{
int res =0;
while (x >0) {
res += c[x];x -=lowbit(x);
}
return res;
}
void add(int x,int d)
{
while (x <=maxnum) {//数组中数字最大值
c[x] += d;x +=lowbit(x);
}
}
int main()
{
int n;
scanf("%d",&n);
for (int i =1; i <= n; i ++) {
scanf("%d",&a[i]);
add(i,a[i]);
}
int l,r;
while (scanf("%d%d",&l,&r) !=EOF) {
printf("%d\n",sum(r) -sum(l - 1));
}
return0;
}
区间更新,区间求和
设原数组第
i
i位的值为
ai
ai,
di=ai−ai−1
di=ai−ai−1,则有(这里认为
a0=0
a0=0):
所以有:
于是我们得到了:
于是我们把原数组差分后维护两个树状数组,一个维护 di di ,一个维护 di×i di×i 。
这样区间求和时可以在两个树状数组中查询得到前缀和,区间修改时就是差分数组的修改,每次修改两个点即可。
void add(int x,int y)
{for(int i=x;i<=n;i+=i&(-i))
c1[i]+=y,c2[i]+=(long long)x*y;}//给差分数组中的位置x加上y
long long sum(int x){//查询前x项的和
long long ans(0);
for(int i=x;i;i-=i&(-i))
ans+=(x+1)*c1[i]-c2[i];
return ans;}
其中
c1i
c1i维护的是
di
di,
c2i
c2i维护的是
di×i
di×i
http://www.cnblogs.com/lcf-2000/p/5866170.html