树状数组和线段树知识点记录

一、树状数组

这个数据结构主要有两个功能，支持单点修改和区间查询。

树状数组其实是个一维的结构，每个数存的都是原数组一段数的和，在单点修改的时候就只需要改变和修改值有关联的几个位置就行。

树状数组有关的几个操作：

（1）lowbit(int x) 返回x的最后一位1

（2）add(int u,int v)在u处加上v （树状数组只能支持加上一个数的操作，如果需要把u处的数i变成j，转换成加上j-i）

（3）query(int x)返回x的前缀和

 

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=100010;
int n,m,w[N],s[N];
int k,a,b;
int lowbit(int x)
{
    return x&-x;
}
int query(int x)
{
    int res = 0;
    for(int i=x;i;i-=lowbit(i)) res+=s[i];
    return res;
}
void add(int u,int v) //在u处加上v
{
    for(int i=u;i<=n;i+=lowbit(i)) s[i]+=v;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) {//树状数组下标必须从1开始
        scanf("%d",&w[i]);
        add(i,w[i]); // 这里用作创建数组
    }
    while(m--)
    {
        scanf("%d%d%d",&k,&a,&b);
        if(k==0) printf("%d\n",query(b)-query(a-1));//前缀和思想
        else add(a,b);
    }
}

二、线段树

与树状数组不同，线段树是个二叉树，二维结构，所以开数组的时候要开四倍大小。

同样支持单点修改和区间查询操作。因为是用结构体实现的，更改结构体也可以达到求区间最大值的作用。 

注：对于下标为u的节点，左子树为u<<1，右子树为u<<1|1。

 三个操作：

（1）pushup(int u)用子节点信息向上传递，更新整个树的信息。

（2）build(int u,int l,int r) 在一段区间上初始化线段树，其中u表示根结点，l表示左边界，r表示右边界

（3）query(u,l,r) 查询某段区间的和，其中u表示根结点，l表示左边界，r表示右边界

（4）modify(u,x,v):修改操作，在u结点中，x位置加上v

树状数组例题的线段树写法：

#include <iostream>
using namespace std;
const int N=100010;
int n,m;
int w[N];
struct node
{
    int l, r;
    int sum;
}tr[N*4];
void pushup(int u)
{
    tr[u].sum = tr[u<<1].sum+tr[u<<1|1].sum;
}
void build(int u,int l,int r)
{
    if(l==r) tr[u] = {l,r,w[l]};
    else 
    {
        tr[u] = {l,r};
        int mid = (l+r)/2;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(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].r == tr[u].l) tr[u].sum+=v;
    else
    {
        int mid = tr[u].r + tr[u].l>>1;
        if(x<=mid) modify(u<<1,x,v);
        else modify(u<<1|1,x,v);
        pushup(u);
    }
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    build(1,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);
    }
}

 

#include <iostream>
#include <climits>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=1e5+10;
int n,m,num[N];
struct node
{
    int l,r,maxv;
}tr[N*4];
void pushup(int u)
{
    tr[u].maxv = max(tr[u<<1].maxv,tr[u<<1|1].maxv);
}
void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,num[l]};
    else
    {
        tr[u] = {l,r};
        int mid = (l+r)>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(u);
    }
}

int query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r) return tr[u].maxv;
    int maxv = INT_MIN;
    int mid = tr[u].l + tr[u].r >>1;
    if(l<=mid) maxv = max(maxv,query(u<<1,l,r));
    if(r>mid) maxv = max(maxv,query(u<<1|1,l,r));
    return maxv;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%d",&num[i]);
    build(1,1,n);
    int l,r;
    while(m--)
    {
        scanf("%d%d",&l,&r);
        printf("%d\n",query(1,l,r));
    }
}