蒟蒻的学习之——树状数组

为什么要用 树状数组？

例如，我们维护数组前缀和s[i]=a[1]+a[2]+……+a[i]，任意修改一个a[i],修改s[i]在最差情况下需要O(n),因此我们需要引入修改与求和都是O(logn)的树状数组。

树状数组最核心的东西

lowbit

lowbit就是二进制的最低位1和它后面的连续0取出来
[代码实现]

int lowbit(int n)
{
    return n&-n;
}

如果你实在搞不懂，那么你就记住就OK啦，或者你可以看一下这篇文章贼易懂的树状数组

单点更新

void update(int x,int val)//更新单节点的值 
{
    while(x<=n)
    {
        tree[x]+=val;
        x+=lowbit(x);
    }
}

区间求和

int sum(int x)//求区间[1,i]的和 
{
    int ans=0;
    while(x>0)
    {
        ans+=tree[x];
        x-=lowbit(x);
    }
    return ans;
}

那么[a,b]之间的和：sum(b)-sum(c-1)

洛谷上的两道模板题

树状数组模板一链接: [https://www.luogu.org/problemnew/show/P3374]

1. 第一题题面

已知一个数列，你需要进行下面两种操作：
1.将某一个数加上x
2.求出某区间每一个数的和

主要还是考察单点修改和区间求和
下面贴AC 代码

#include<bits/stdc++.h>
#define R register int
#define maxn 500005 
using namespace std;
int n,m,tree[maxn];
int lowbit(int n)
{
    return n&-n;
}
void update(int x,int val)//更新单节点的值 
{
    while(x<=n)
    {
        tree[x]+=val;
        x+=lowbit(x);
    }
}
int sum(int x)//求区间[1,i]的和 
{
    int ans=0;
    while(x>0)
    {
        ans+=tree[x];
        x-=lowbit(x);
    }
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(R i=1;i<=n;++i)
    {
        int a;
        scanf("%d",&a);
        update(i,a);//!!!
    }
    for(R i=1;i<=m;++i)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        if(a==1) update(b,c);
        else printf("%d\n",sum(c)-sum(b-1));	
    }
    return 0;	
}

在原数组的输入的方式上需要注意update(i,a)，是不是很巧妙

   for(R i=1;i<=n;++i)
    {
        int a;
        scanf("%d",&a);
        update(i,a);//
    }

树状数组模板二链接
[https://www.luogu.org/problemnew/show/P3368]

1.第二题题面

已知一个数列，你需要进行下面两种操作：
1.将某区间每一个数数加上x
2.求出某一个数的值

这题主要是需要考虑差分
令b[i]=a[i]-a[i-1],a[0]=0,那么a[i]=b[1]+b[2]+b[3]+…+b[i]
b[ ]数组就是a[ ]数组的差分数组
对区间[x,y]做修改，只需要修改b[x]与b[y+1]
b[x]=b[x]+k,b[y+1]=b[y+1]-k
贴部分代码

int main()
{
    scanf("%d%d",&n,&m);
    int a,now=0;
    for(R i=1;i<=n;++i)
    {
        scanf("%d",&a);
        update(i,a-now);
        now=a;
    }
    for(R i=1;i<=m;++i)
    {
        int x; 
        scanf("%d",&x);
        if(x==1)
        {
            int l,r,k;
            scanf("%d%d%d",&l,&r,&k);
            update(l,k);
            update(r+1,-k);
        }
        else
        {
            int k;
            scanf("%d",&k);
            printf("%d\n",sum(k));
        }	
    }
    return 0;	
}