蒟蒻成长日记 7.24 树状数组的基础拓展
树状数组
———————短小精悍的手术刀
可以解决的基本问题:
1.单点更新和单点查询操作
2.单点更新和区间查询操作
3.区间更新和单点查询操作
4.区间查询和区间更新操作
上一篇博客简单的介绍了树状数组的初步露面,代码上对应的也比较简单,只介绍了树状数组的裸模板,下面通过题目简单讨论一下树状数组的的简单应用也就是支持的3,4操作
注:裸树状数组的基本函数有:add(int x,int c)//在x位置加上c,sum(int x)//返回[1,x]的区间和,lowbit(int x)
3.区间更新和单点查询操作
洛谷3374
题目描述
如题,已知一个数列,你需要进行下面两种操作:
将某一个数加上 x
求出某区间每一个数的和
输入格式
第一行包含两个正整数 n,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 n 个用空格分隔的整数,其中第 i 个数字表示数列第 i 项的初始值。
接下来 m 行每行包含 3 个整数,表示一个操作,具体如下:
1 x k 含义:将第 x 个数加上 k
2 x y 含义:输出区间 [x,y] 内每个数的和
输入输出样例
输入样例:
5 5
1 5 4 2 3
1 2 4 2
2 3
1 1 5 -1
1 3 5 7
2 4
输出样例:
6
10
思路:首先我们裸的树状数组是支持的单点更新和区间查询操作,复杂度是log级别,本题要求我们解决的是区间查询和单点更新操作,不难想到我们在学前缀和的时候也学过一个差分来解决区间查询操作,本题加一个差分,把树状数组构造成差分数组就可以了,这也是树状数组的一个简单模板。
#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
int n,m;
int q[N];
int tr[N];
int lowbit(int x)
{
return x&-x;
}
void add(int x,int c)
{
for(int i=x;i<=n;i+=lowbit(i))tr[i]+=c;
}
int sum(int x)
{
int res=0;
for(int i=x;i;i-=lowbit(i))res+=tr[i];
return res;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)scanf("%d",&q[i]);
for(int i=1;i<=n;i++)add(i,q[i]-q[i-1]);
while(m--)
{
int a,b,c,d;
scanf("%d",&a);
if(a==1)
{
scanf("%d%d%d",&b,&c,&d);
add(b,d),add(c+1,-d);
}
else
{
scanf("%d",&c);
printf("%d\n",sum(c));
}
}
}
4.区间查询和区间更新操作
一个简单的整数问题2:acwing 243题
题意:
给定一个长度为 NN 的数列 AA,以及 MM 条指令,每条指令可能是以下两种之一:
C l r d
,表示把 A[l],A[l+1],…,A[r]A[l],A[l+1],…,A[r] 都加上 dd。Q l r
,表示询问数列中第 l∼rl∼r 个数的和。
对于每个询问,输出一个整数表示答案。
输入格式
第一行两个整数 N,MN,M。
第二行 NN 个整数 A[i]A[i]。
接下来 MM 行表示 MM 条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4
输出样例:
4
55
9
15
思路:推公式加差分
有了思路,代码就是在模板上改了一下
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long LL;
LL tr1[N],tr2[N];
int n,m;
int q[N];
int lowbit(int x)
{
return x&-x;
}
void add(int x,LL tr[],LL c)
{
for(int i=x;i<=n;i+=lowbit(i))tr[i]+=c;
}
LL sum(LL x)
{
LL res1=0,res2=0;
for(int i=x;i;i-=lowbit(i))res1+=tr1[i],res2+=tr2[i];
return x*res1-res2;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)scanf("%d",&q[i]);
for(int i=1;i<=n;i++)add(i,tr1,q[i]-q[i-1]),add(i,tr2,(LL)(i-1)*(q[i]-q[i-1]));
char op[2];
LL a,b,d;
while(m--)
{
scanf("%s%lld%lld",op,&a,&b);
if(*op=='Q')
printf("%lld\n",sum(b)-sum(a-1));
else
{
scanf("%lld",&d);
add(a,tr1,d),add(b+1,tr1,-d);
add(a,tr2,(a-1)*d),add(b+1,tr2,b*(-d));
}
}
}
参考资料:
acwing算法提高课 树状数组
CSDN 博主Ac_dream 树状数组_AC__dream的博客-CSDN博客