一、树状数组介绍:
(1)一般来说,树状数组用来维护一个数组的前缀和,这个数组可以是差分数组,也可以是原数组;除此之外,树状数组延申出来另一个性质,可以表示数组中某个数存在或不存在。
(2)用树状数组维护的数组下标一般从1开始
二、单点修改,区间查询:
单点修改区间查询维护的是原数组
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int t[N],a[N],n,m;
int lowbit(int x){
return x&-x;
}
void add(int x,int val){
for(int i=x;i<=n;i+=lowbit(i)) t[i]+=val;
}
int query(int x){
int res;
for(int i=x;i;i-=lowbit(i)) res+=t[i];
return res;
}
int main()
{
cin >> n >> m;
for(int i=1;i<=n;i++){
cin >> a[i];
add(i,a[i]);
}
while(m--){
int l,r;
cin >> l >> r;
cout << query(r)-query(l-1);
}
return 0;
}
三、区间更新,单点查询:
区间更新,单点查询维护的是差分数组
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int t[N],a[N],n,m;
int lowbit(int x){
return x&-x;
}
void add(int x,int val){
for(int i=x;i<=n;i+=lowbit(i)) t[i]+=val;
}
int query(int x){
int res;
for(int i=x;i;i-=lowbit(i)) res+=t[i];
return res;
}
int main()
{
cin >> n >> m;
for(int i=1;i<=n;i++){
cin >> a[i];
add(i,a[i]-a[i-1]);
}
while(m--){
int x;
cin >> x;
cout << query(x);
}
return 0;
}
四、区间修改,区间查询:
区间修改,区间查询需要维护两个差分数组的前缀和,这两个数组表示的含义是由公式推导出来的
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int sum1[N];//sum1维护(d1+d2+...dn)的差分数组的前缀和
int sum2[N];//sum2维护(0*d1+1*d2+...+(n-1)*dn)的差分数组的前缀和
int a[N],n,m;
int lowbit(int x){
return x&-x;
}
void add(int x,int val){
for(int i=x;i<=n;i+=lowbit(i)){
sum1[i]+=val;
sum2[i]+=(x-1)*val;
}
}
int query(int x){
int res;
for(int i=x;i;i-=lowbit(i)) res+=x*sum1[i]-sum2[i];
return res;
}
int main()
{
cin >> n >> m;
for(int i=1;i<=n;i++){
cin >> a[i];
add(i,a[i]-a[i-1]);
}
while(m--){
int l,r;
cin >> l >> r;
cout << query(r)-query(l-1);
}
return 0;
}
五、树状数组求顺序对与逆序对:
(1)树状数组求顺序对与逆序对需要保证已经是按下标值排好序的
(2)树状数组求顺序对和逆序对需要尽量保证a数组里面的数从1开始,并连续,否则的话,一个值对应树状数组的一个下标,会很浪费空间,如果出现数组a很稀疏的情况,就要进行离散化
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int t[N];//此时树状数组t的含义是每个数有没有(1或0)
int a[N],n,m;
int lowbit(int x){
return x&-x;
}
void add(int x,int val){
for(int i=x;i<=n;i+=lowbit(i)) t[i]+=val;
}
int query(int x){
int res;
for(int i=x;i;i-=lowbit(i)) res+=t[i];
return res;
}
int main()
{
cin >> n >> m;
for(int i=1;i<=n;i++){//按照顺序添加和询问表示顺序对,逆序的话就是求逆序对
cin >> a[i];
cout << query(a[i]-1);//这个query的含义就是当前的a[i]的顺序对有多少对
add(a[i],1);//往a[i]这个位置加上1,表示a[i]存在
}
return 0;
}
放一道题目: AcWing 241. 楼兰图腾 - AcWing (已做笔记)
六、由顺序对的对数倒推原数组:
我们一开始可以初始化树状数组中每个数都存在(赋值为1),然后倒序遍历顺序对数组,a[i]表示第a[i]+1小的数,所以我们只要在树状数组里面找到剩余的第a[i]+1小的数即可,这一步我们可以用二分操作(详情见打卡里面的代码)
放一道题目: AcWing 244. 谜一样的牛 - AcWing (已做笔记)