树状数组全解析---一遍必会的教程

树状背包

对于求前缀和以及修改区间值 时间复杂度的一个平衡—将两个操作的时间复杂度都降低至O(logn)

算法原理

基本模板

查询操作

对想求【1，x】的区间和

这里给出x的二进制形式 x = 0100101

我们把【1，x】划分为以下区间

(0100100，0100101】

(0100000，0100100】

…

这里定义 tri[x] 为以x为右端点，长度为lowbit（x）的区间和

//在这样的操作下 x的区间和代码为
int sum(int x)
{
    int res = 0;
	for(int i=x;i;i-=lowbit(i))
    	res += tri[i];
	return res;
}

改值操作

首先我们通过一张图看一下tri数组之间的关系

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-0AJsVt8o-1598101994608)(C:\Users\lin\Desktop\IMG_0097.PNG)]

可以观察到每一个tri数组的父节点只有一个—恰好是树的结构—这也是树状数组名字的由来

所以修改一个点他所直接影响的tri节点只有一个—即 x + lowbit(x)-----（至于原因–可以想想怎么由父节点拆分出所有子节点）

可以得到修改代码如下

void add(int x,int u)
{
	for(int i=x;i<=n;i+=lowbit(i))
		tri[i] += x;
}

有了上面两个操作—我们就完成了最简单的树状数组的模板了

模板题—洛谷P3374 【模板】树状数组 1

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int N = 1e6;
int tri[N];
int n,m;
inline int lowbit(int x)
{
    return x&-x;
}

void add(int x,int u)
{
    for(int i=x;i<=n;i+=lowbit(i))
        tri[i]+=u;
}

int sum(int x)
{
    int ans = 0;
    for(int i=x;i;i-=lowbit(i))
        ans += tri[i];
    return ans;
}

int main()
{
    cin>>n>>m;
    int a,b,c;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a);
        add(i,a);
    }

    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&c,&a,&b);
        if(c==1)
        {
            add(a,b);
        }
        else
        {
            cout<<sum(b) - sum(a-1)<<endl;
        }
    }
    return 0;
}

扩展应用

树状数组和差分的联系

模板 P3368 【模板】树状数组 2

​ 如题，已知一个数列，你需要进行下面两种操作：

1. 将某区间每一个数数加上 x；
2. 求出某一个数的值。

只需让树状数组每个点存差分数组即可----求区间和【1，x】即是求x的值

//修改区间内每个点的值----查询单点值----树状数组套差分数组
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int N = 6e5;
int tri[N];
int n,m;
int num[N];
inline int lowbit(int x)
{
    return x&-x;
}

inline void add(int x,int u)
{
    for(int i=x;i<=n;i+=lowbit(i))
        tri[i] += u;
}

inline int sum(int x)
{
    int ans = 0;
    for(int i=x;i;i-=lowbit(i))
        ans += tri[i];
    return ans;
}

int main()
{
    cin>>n>>m;
    int a,b,c,d;

    for(int i=1;i<=n;i++)
        scanf("%d",&num[i]);
    
    for(int i=1;i<=n;i++)
        add(i,num[i]-num[i-1]);
    
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&c);
        if(c==2)
        {
            scanf("%d",&a);
            cout<<sum(a)<<endl;
        }
        else
        {
            scanf("%d%d%d",&a,&b,&d);
            add(a,d);
            add(b+1,-d);
        }
    }
    return 0;
}

树状数组加上数学公式

POJ 3468

“C a b c” means adding c to each of Aa, Aa+1, … , Ab. -10000 ≤ c ≤ 10000.
“Q a b” means querying the sum of Aa, Aa+1, … , Ab.

/*
    要求求区间和---并且修改区间内的每一个值

    做法1.线段树
    做法2.树状数组加公式优化

    tri[i] 存差分数组
    [1,x] = (tri[1]+...+tri[x])*n - tri[1]*(1-1) - tri[2]*(2-1)+.....
    故建立两个树状数组---一个存a[i] 一个存a[i]*(i-1); 
    sum(tri1,x) - sum(tri2,x)即为所求

    修改---
*/
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int N = 2e5;
typedef long long ll;
ll tri1[N];
ll tri2[N];
int n,m;
int num[N];

inline int lowbit(int x)
{
    return x&-x;
}

inline void add(ll tri[],int x,ll u)
{
    for(int i=x;i<=n;i+=lowbit(i))
        tri[i] += u;
}

inline ll sum(ll tri[],int x)
{
    ll ans = 0;
    for(int i=x;i;i-=lowbit(i))
        ans += tri[i];
    return ans;
}


inline ll fsum(int x)
{
    return (ll)sum(tri1,x)*(x) - sum(tri2,x);
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        scanf("%d",&num[i]);
    
    for(int i=1;i<=n;i++)
        add(tri1,i,(num[i]-num[i-1])),add(tri2,i,(ll)(num[i]-num[i-1])*(i-1));
    
    char ch[2];
    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        scanf("%s",ch);
        if(ch[0]=='Q')
        {
            scanf("%d%d",&a,&b);
            cout<<fsum(b) - fsum(a-1)<<endl;
        }
        else
        {
            scanf("%d%d%d",&a,&b,&c);
            add(tri1,a,c),add(tri1,b+1,-c);
            add(tri2,a,(ll)c*(a-1)),add(tri2,b+1,(ll)-c*b);
        }
    }
    return 0;
}