树状数组 :
树状数组就是利用二进制的巧妙创建出了效率更高的求前缀和的方法
upadate 函数向上更新 , query 函数查询前缀和
实现方法:假设已经创建好了树状数组,此时需要在位置5添加增加数值10。
5 二进制 101 , 在最后的1的位置加上一个1 ,就得到110 再加 就得 1000
101 , 110 , 1000 ,等 上加上10 ,此时整个树状数组就更新好了。
查询 : 需要查询位置22的值 。
22 二进制 10110 , 依次抹去最后的1,可得到10100,10000,00000,
因为在00000上是没有值的,所以只需加上 10110,10100,10000上的值,他们的和就是22的前缀和。
修改,查询一个数的时间复杂度为O(log2 n),树状数组解决前缀和的问题,也可以很好的处理修改整个区间的值,并查询某一个值,利用差分前缀和便可以解决。
上实战:
如题,已知一个数列,你需要进行下面两种操作:
-
将某一个数加上 xx
-
求出某区间每一个数的和
输入格式
第一行包含两个正整数 n,m分别表示该数列数字的个数和操作的总个数。
第二行包含 n个用空格分隔的整数,其中第 i个数字表示数列第 i 项的初始值。
接下来 m行每行包含 3 个整数,表示一个操作,具体如下:
接下来 mm 行每行包含 33 或 44 个整数,表示一个操作,具体如下:
输入格式
第一行包含两个整数 n, mn,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 nn 个用空格分隔的整数,其中第 ii 个数字表示数列第 ii 项的初始值。
输入:
-
1 x k
含义:将第 x 个数加上 k -
2 x y
含义:输出区间 [x,y] 内每个数的和 -
输入:
-
5 5 1 5 4 2 3 1 1 3 2 2 5 1 3 -1 1 4 2 2 1 4
- 输出:
-
14 16
- 代码:
-
#include<iostream> #include<stdio.h> #define ll long long using namespace std; ll arr[600000]; ll n,m; ll query(ll x) { ll sum=0; for(;x;x-=x&-x) sum+=arr[x];//x&-x 可以求得最后的一 return sum; } void update(ll x,ll a) { for(;x<=n;x+=x&-x) arr[x]+=a; } int main() { cin>>n>>m; for(int i=1;i<=n;i++) { ll x;scanf("%lld",&x); update(i,x); } while(m--) { ll choose; scanf("%lld",&choose); if(choose==1) { ll x,k; scanf("%lld%lld",&x,&k); update(x,k); } else { ll x,y; scanf("%lld%lld",&x,&y); cout<<query(y)-query(x-1)<<endl; } } return 0; }
-
线段树:
-
对于区间修改,区间查询,线段树或许会更稳
-
通过二分的方法向下一直分,直到分到单个为止,分的每一个区块有个值代表整个区间的和
-
创建结构体,存储左右端点的值,还需要存这个区间的和
-
struct Node
{
ll l,r,sum,add;
}tr[N*4]; -
结构体中多存了个add,此时的add为懒标记,记录此时的区块的以下区间的每一个数需要加上add,在向下更新时就可以用到
-
在最开始时需要创建整个树,利用结构体来存储。
-
tr[1] 为最开始的整个区块。
-
tr[1<<1] 为tr[1]的左区块,tr[1<<1|1] 为tr[1]的右区块1<<1 等价于 1*2 ; 1<<1|1 等价于 1*2 + 1
-
创建树的代码
-
void build(ll u,ll l,ll r)
{
if(l==r) tr[u]={l,r,w[r],0};
else
{
tr[u]={l,r,0,0};
ll mid=(l+r)>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
pushup(u);
}
} -
升级的代码,爸爸的财富等于两个儿子的财富
-
void pushup(ll u)
{
tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
} -
向下更新的代码,爸爸的add中有值,说明爸爸赚了钱,爸爸的钱就是儿子的钱,爸爸赚了多少,儿子就赚了多少。
-
void pushdown(ll u)
{
auto &root=tr[u],&left=tr[u<<1],&right=tr[u<<1|1];
if(root.add)
{
left.add+=root.add,left.sum+=(left.r-left.l+1)*root.add;
right.add+=root.add,right.sum+=(right.r-right.l+1)*root.add;
root.add=0;
}
} -
对于一段区间的添加值
-
void modify(ll u,ll l,ll r,ll d)
{
if(l<=tr[u].l&&tr[u].r<=r) // 找到在它之内的区间
{
tr[u].sum+=(tr[u].r-tr[u].l+1)*d;
tr[u].add+=d;
return ;
}
pushdown(u); // 向下更新,保证左右的值是正确的
ll mid=(tr[u].l+tr[u].r)>>1;
if(l<=mid) modify(u<<1,l,r,d);
if(r>mid) modify(u<<1|1,l,r,d);
pushup(u); // 出来时更新爸爸的值
} -
查询每一段区间:
-
ll query(ll u,ll l,ll r)
{
if(l<=tr[u].l&&tr[u].r<=r) return tr[u].sum;
pushdown(u); //向下更新 , 保证左右的值是正确的
ll sum=0;
ll mid=(tr[u].l+tr[u].r)>>1;
if(l<=mid) sum+=query(u<<1,l,r);
if(r>mid) sum+=query(u<<1|1,l,r);
return sum;
} -
上实战:
-
如题,已知一个数列,你需要进行下面两种操作:
- 将某区间每一个数加上 kk。
- 求出某区间每一个数的和。
1 x y k
:将区间 [x, y][x,y] 内每个数加上 kk。2 x y
:输出区间 [x, y][x,y] 内每个数的和。
5 5 1 5 4 2 3 2 2 4 1 2 3 2 2 3 4 1 1 5 1 2 1 4
输出:
11 8 20
代码:s
#include<iostream>
#include<stdio.h>
#define ll long long
using namespace std;
const int N=1e5;
ll w[N+5];
ll n,m;
struct Node
{
ll l,r,sum,add;
}tr[N*4];
void pushup(ll u)
{
tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void pushdown(ll u)
{
auto &root=tr[u],&left=tr[u<<1],&right=tr[u<<1|1];
if(root.add)
{
left.add+=root.add,left.sum+=(left.r-left.l+1)*root.add;
right.add+=root.add,right.sum+=(right.r-right.l+1)*root.add;
root.add=0;
}
}
void build(ll u,ll l,ll r)
{
if(l==r) tr[u]={l,r,w[r],0};
else
{
tr[u]={l,r,0,0};
ll mid=(l+r)>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
pushup(u);
}
}
void modify(ll u,ll l,ll r,ll d)
{
if(l<=tr[u].l&&tr[u].r<=r)
{
tr[u].sum+=(tr[u].r-tr[u].l+1)*d;
tr[u].add+=d;
return ;
}
pushdown(u);
ll mid=(tr[u].l+tr[u].r)>>1;
if(l<=mid) modify(u<<1,l,r,d);
if(r>mid) modify(u<<1|1,l,r,d);
pushup(u);
}
ll query(ll u,ll l,ll r)
{
if(l<=tr[u].l&&tr[u].r<=r) return tr[u].sum;
pushdown(u);
ll sum=0;
ll mid=(tr[u].l+tr[u].r)>>1;
if(l<=mid) sum+=query(u<<1,l,r);
if(r>mid) sum+=query(u<<1|1,l,r);
return sum;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) scanf("%lld",&w[i]);
build(1,1,n);
for(int i=1;i<=m;i++)
{
ll choose;
scanf("%lld",&choose);
if(choose==1)
{
ll x,y,k;
scanf("%lld%lld%lld",&x,&y,&k);
modify(1,x,y,k);
}
else
{
ll x,y;
scanf("%lld%lld",&x,&y);
cout<<query(1,x,y)<<endl;
}
}
return 0;
}