树状数组总结

一、树状数组介绍：

（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  （已做笔记）