第五讲---树状数组与线段树（上课）

1.树状数组

树状数组就是c[x]，存的值就是原数组中(下标从1开始)[x-lowbit(x),x]这一段区间的前缀和。

1.1动态求连续区间和

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n, m;
int a[N], tr[N];
//下面三个为树状数组的三个操作，要熟记。 
int lowbit(int x)
{
    return x & -x;
}

void add(int x, int v)//当原数组某一个数加上v,树状数组也要进行add操作，才能保证用树状数组求出来的前缀和依然正确 
{
    for (int i = x; i <= n; i += lowbit(i)) tr[i] += v;
}

int query(int x)//求原数组的前缀和 
{
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    for (int i = 1; i <= n; i ++ ) add(i, a[i]);//构造树状数组，a,tr数组最开始都为空，看成 不断给a数组加上a[i],要保证前缀和不求错，就要给c数组 add(i, a[i])

    while (m -- )
    {
        int k, x, y;
        scanf("%d%d%d", &k, &x, &y);
        if (k == 0) printf("%d\n", query(y) - query(x - 1));//用树状数组求原数组的前缀和
        else add(x, y);
    }

    return 0;
}

树状数组只能处理加号，如果不是加号，想办法变成加号

1.2数星星

此题考查树状数组的应用

1. 题目要求求某一个点(x,y)左下方星星的个数(不包括自己)，且星星按y坐标增序给出，y 坐标相同的按x坐标增序给出,因此对于每个新来的点(x,y),y是当前纵坐标的最大值，只需要求[1,x]中星星出现的数量即可

2. 通过树状数组完成单点修改，区间查询操作

注意：树状数组是从1开始的，而题目的给定的x范围是0≤x≤32000,因此需要将所有的x赋值成x + 1(相对位置不变)

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 32010;    //横坐标

int n;
int tr[N];
int level[N];//用于记录在第几个等级的星星数量  下标表示等级

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

void add(int x)
{
    //这里需要注意 不能 <= n 我太菜了 一开始设置为n 出错了  下标代表的是 x坐标的范围 所以要根据题目给的
     
    for(int i = x;i <= 32001;i += lowbit(i))    tr[i]++;
    //tr树状数组   每个位置上的值 表示 在该x坐标的星星数 加上它 下级的星星数
}

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

int main()
{
    scanf("%d",&n);
    for(int i = 1;i <= n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);//其实这里 y 并没有什么用
        //树状数组从1开始
        x++;/*为了防止出现0的情况，给它全体横坐标加上 1 就好了。
         这其实是一个很小的细节，作者但是做的时候没考虑到然后就wa了，而给每个 x 都加上 1 并不会影响结果*/
        level[sum(x)] ++;/*然后查一下它的前缀和是多少，前缀和是多少就意味着是多少级
        这是一个动态变化的过程，而且后面的一定比前面高
         所以要实时计算*/
        add(x);//把x位置的星星个数 +1
    }
    for(int i = 0;i < n;i++)    printf("%d\n",level[i]);//输出每一个等级的数量
    
    return 0;
}

2.线段树

就是二叉树，用于单点修改，区间查询。
比较容易理解，适用性较高，但是代码复杂


结点个数

2.1动态求连续区间和

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>

using namespace std;

const int N=100010;

int n,m;

int w[N];//记录一下权重

struct node
{
    int l,r;//左右区间
    int sum;//总和
}tr[N*4];//记得开 4 倍空间

void push_up(int u)//利用它的两个儿子来算一下它的当前节点信息
{
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;//左儿子 u<<1 ,右儿子 u<<1|1  
}

void build(int u,int l,int r)/*第一个参数，当前节点编号，第二个参数，左边界，第三个参数，右边界*/
{
    if(l==r)tr[u]={l,r,w[r]};//如果当前已经是叶节点了，那我们就直接赋值就可以了
    else//否则的话，说明当前区间长度至少是 2 对吧，那么我们需要把当前区间分为左右两个区间，那先要找边界点
    {
        tr[u]={l,r};//这里记得赋值一下左右边界的初值

        int mid=l+r>>1;//边界的话直接去计算一下 l + r 的下取整

        build(u<<1,l,mid);//先递归一下左儿子

        build(u<<1|1,mid+1,r);//然后递归一下右儿子

        push_up(u);//做完两个儿子之后的话呢 push_up 一遍u 啊，更新一下当前节点信息
    }
}

int query(int u,int l,int r)//查询的过程是从根结点开始往下找对应的一个区间
{
    if(l<=tr[u].l&&tr[u].r<=r)return tr[u].sum;//如果当前区间已经完全被包含了，那么我们直接返回它的值就可以了
    //否则的话我们需要去递归来算
    int mid=tr[u].l+tr[u].r>>1;//计算一下我们 当前 区间的中点是多少
    //先判断一下和左边有没有交集

    int sum=0;//用 sum 来表示一下我们的总和

    if(mid>=l)sum+=query(u<<1,l,r);//看一下我们当前区间的中点和左边有没有交集
    if(r>=mid+1)//看一下我们当前区间的中点和右边有没有交集
    sum+=query(u<<1|1,l,r);

    return sum;
}

void modify(int u,int x,int v)//第一个参数也就是当前节点的编号,第二个参数是要修改的位置，第三个参数是要修改的值
{
    if(tr[u].l==tr[u].r)tr[u].sum+=v; //如果当前已经是叶节点了，那我们就直接让他的总和加上 v 就可以了

    //否则
    else
    {

      int mid=tr[u].l+tr[u].r>>1;
      //看一下 x 是在左半边还是在右半边
      if(x<=mid)modify(u<<1,x,v);//如果是在左半边，那就找左儿子
      else modify(u<<1|1,x,v);//如果在右半边，那就找右儿子

      //更新完之后当前节点的信息就要发生变化对吧，那么我们就需要 pushup 一遍

      push_up(u);
    }

}

int main()
{
    scanf("%d%d",&n,&m);

    for(int i=1;i<=n;i++)scanf("%d",&w[i]);

    build(1,1,n);/*第一个参数是根节点的下标，根节点是一号点，然后初始区间是 1 到 n */

    //后面的话就是一些修改操作了

    while(m--)
    {
        int k,a,b;

        scanf("%d%d%d",&k,&a,&b);

        if(k == 0)printf("%d\n",query(1,a,b));//求和的时候，也是传三个参数，第一个的话是根节点的编号 ，第二个的话是我们查询的区间 
        //第一个参数也就是当前节点的编号
        else
        modify(1,a,b);//第一个参数是根节点的下标,第二个参数是要修改的位置，第三个参数是要修改的值

    }


    return 0;
}

2.2数列区间最大值