树状数组:
一. 作用:
1. 处理单点修改以及前缀查询
2. 特殊情况的区间修改与区间查询
二. 复杂度
修改: O(logn)
查询: O(logn)
空间: O(n)
三. 具体实现
生成
C[i]=Σa[j] (i-2^k<j<=i)
j ∈ (i-lowbit(i) , i);
lowbit(i) = (x & (-x))
int lowbit(int x) {return x&-x;}
int work(int x) {int sum=0; for (;x;x-=lowbit(x)) sum+=C[x]; return sum;}
int Query(int l,int r) {return sum(r)-sum(l-1);}
C[1]=a[1];
C[2]=a[1]+a[2];
C[3]=a[3];
C[4]=a[1]+a[2]+a[3]+a[4];
C[5]=a[5];
C[6]=a[5]+a[6];
C[7]=a[7];
C[8]=a[1]+a[2]+a[3]+a[4]+a[5]+a[6]+a[7]+a[8];
当ai增加x时如何修改C
(1) 令C[i]+=x。
(2) i+=lowbit(i)。
(3) 若i>n,退出,否则执行(1)。
void change(int i,int x){
for (; i<=n; i+=lowbit(i)) C[i]+=x;
}
区间增加,单点查询
令b[i]=a[i]-a[i-1]。
a[x]=∑b[j] (j<=x)
修改复杂度:O(1),查询复杂度(对b维护一个前缀和):O(1)
区间增加,区间查询和
b数组相关与上题相同
ans=(Σb[i])(i<=l)*(r-l+1)-Σ(b[i]i)(l<i<=r)+(Σb[i])(l<i<=r)(r+1)
int a[10010];//数据
int b[10010];//a[i]-a[i-1]
int c[10010];//树状数组核心
int lowbit(int x) {
return x&-x;
}
void update(int x) {//更新单个c[x]
c[x]=0;
for(int j=x-lowbit(x);j<=x;j++){
c[x]+=a[j];
}
}
void change(int i,int x) {//将a[i]的值增加x,并修改所有相关的c的值
// a[i]+=x;
for(;i<=n;i+=lowbit(i))//n为数据的个数(a的长度)
c[i]+=x;
}
//单点修改,区间查询和 (35~43)
int preSum(int x) {//求前缀和
int sum=0;
for(;x;x-=lowbit(x))
sum+=c[x];
return sum;
}
int query(int l,int r){
return preSum(r)-preSum(l-1);
}
//区间增加,单点查询 (46~62)
int preB[10010];
void init() {
preB[0]=0;
for(int i=1;i<=n;i++){
b[i]=a[i]-a[i-1];
preB[i]=preB[i-1]+b[i];
}
}
void modify(int l,int r,int v) {//[l,r]+v
b[l]+=v;
b[r+1]-=v;
for(int i=l;i<r+1;i++)
preB[i]+=v;
}
int query(int x) {///查询a[x]
return preB[x];
}
//区间增加,区间查询和 (65~87)
int preB[10010];
void init() {
preB[0]=0;
for(int i=1;i<=n;i++){
b[i]=a[i]-a[i-1];
preB[i]=preB[i-1]+b[i];
}
}
void modify(int l,int r,int v) {//[l,r]+v
b[l]+=v;
b[r+1]-=v;
for(int i=l;i<r+1;i++)
preB[i]+=v;
}
int bii(int l,int r){
int ans=0;
for(int i=l;i<r+1;i++)
ans+=b[i]*i;
return ans;
}
int query(int l,int r) {
return preB[l]*(r-l+1) - bii(l+1,r) + (preB[r+1]-preB[l]) * (r+1);
}