树状数组可以办到
- 区间查询
- 单点查询
- 单点修改
就是做不到区间修改
但是呢,区间修改,你完全完全可以直接用树状数组的单点修改去做,就是时间复杂度会稍微慢了一丢丢。无所谓的。
但是!!这里有一个可以实现和线段树一样快的树状数组的方法!!
树状数组就是一个前缀和,树状数组和前缀和没有任何差别,就是当你需要对于一个数组维护,一边修改,修改完再查前缀和,需要这样的操作的时候你再去使用树状数组。这样会比较节省时间和空间,要是你只有看前缀和的话,你还是就直接用前缀和算了。
前缀和呢,其实也是完全可以看成是可以求一段区间很好求的东西。
其中有且只有3个很简单的函数
lowbit
函数,求x的二进制表达式中最低位的的1所对应的值x&(-x)
add
函数,参数有2
个,x,v
表示第x
个数加上v
循环,从这个数到最大是数N
(这个要特别注意一下),里面就是正常的加哦tr[i] += v
注意,add函数的循环和query函数的循环每一步的幅度都是lowbit(i)
,注意这里lowbit(i)实战中背错一次!!!sum
函数,参数就1
个x
,表示前x
个数相加的和
循环,i = x
到i(i>0)
变小的循环,res += tr[i]
虽然是前缀和,但是却不是+=a[i]
哦,这里是树状数组领域!
注意点
- 你需要两个数组,一个是
a
存的是你每一个的数字,一个是你的前缀和数组tr
,因为名字叫树状数组,所以也就起一个tr这个好听的名字(tree) - 1.读入每一个数字2.
add(i,a[i])
这两步是连续进行的,注意,下标都是从1
开始,这一点很重要,下标必须从1开始,如果他的输入是有可能从0开始的,那你就x++ - 有一个巨重要巨重要的点,就是你输入a[i]的时候要同时进行add(i,a[i])操作!!!!千万不要忘记了!!!
- 对于树状数组添加的维护,一开始就要进行!!!
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int n, m;
int a[N], tr[N];
int lowbit(int x)
{
return x & -x;
}
void add(int x, int v)
{
for (int i = x; i <= N; i += lowbit(i)) tr[i] += v;
}
int sum(int x)
{
int res = 0;
for (int i = x; i; i -= lowbit(i)) res += tr[i];
return res;
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
for (int i = 1; i <= n; i ++ ) add(i, a[i]);
while (m -- )
{
int k, x, y;
scanf("%d%d%d", &k, &x, &y);
if (k == 0) printf("%d\n", sum(y) - sum(x - 1));
else add(x, y);
}
return 0;
}
这,就是可以实现和线段树一样的快速区间修改的树状数组!
#include<bits/stdc++.h>
using namespace std;
long long n,m,a[200010],b[200010],c[200010],bb[200010],c1[200010];
long long lowbit(long long x)
{
return x&-x;
}
void modify(int x,long long y)
{
for (int i=x;i<=n;i+=lowbit(i))
b[i]+=y;
}
long long query(long long x)
{
long long ans=0;
for (int i=x;i>0;i-=lowbit(i))
ans+=b[i];
return ans;
}
void modifyy(long x,long long y)
{
for (int i=x;i<=n;i+=lowbit(i))
bb[i]+=y;
}
long long queryy(long long x)
{
long long ans=0;
for (int i=x;i>0;i-=lowbit(i))
ans+=bb[i];
return ans;
}
int main()
{
cin>>n>>m;
for (int i=1;i<=n;i++)
{
cin>>a[i];
c[i]=a[i]-a[i-1];
c1[i]=(i-1)*c[i];
}
for (int i=1;i<=n;i++)
modify(i,c[i]);
for (int i=1;i<=n;i++)
modifyy(i,c1[i]);
for (int i=1;i<=m;i++)
{
long long p;
cin>>p;
if (p==1)
{
long long xx,yy,zz;
cin>>xx>>yy>>zz;
modify(xx,zz);
modify(yy+1,-zz);
modifyy(xx,(xx-1)*zz);
modifyy(yy+1,-yy*zz);
}
if (p==2)
{
long long xx,yy,ss;
cin>>xx>>yy;
ss=(yy*query(yy)-queryy(yy))-((xx-1)*query(xx-1)-queryy(xx-1));
cout<<ss<<endl;
}
}
return 0;
}