线段树学习笔记

线段树：
线段树其实是一个二叉搜索树，它存储的是一个区间的信息
每个节点以结构体的方式存储

结构体包括区间的左右端点外加区间和
其它看情况加区间要维护的信息

线段树的基本思想是二分

特殊性质有：
每个节点的左孩子的区间范围确定
对于节点k，左孩子节点为2k，右孩子为2k+1，这符合完全二叉树的性质

线段树的基本操作：建树，单点查询，单点修改，区间查询，区间修改

1.建树

struct node
{
    int l,,t;
    int sum;//l，r分别表示区间左右端点 sum表示区间和
}tree[4*n+1];

树应该开4倍空间
主体思路

对于二分到的结点，给他左右端点确定范围
如果是叶子结点，存储要维护的信息
状态合并

void creat_tree(int l,int r,int k)
{
    tree[k].l=l;tree[k].r=r;
    if(l==r)
    {
        cin>>tree[k].sum;
        return;
    }
    int m=(l+r)/2;
    creat_tree(l,m,k*2);
    creat_tree(m+1,r,k*2+1);
    tree[k].sum=tree[2*k].sum+tree[k*2+1].sum;
}

2.单点查询
如果左端点等于右端点则是叶节点，是我们要查找的值，直接返回该值即可
如果不相等则求出其左右端点的中间值，如果比mid小递归左孩子，否则递归右孩子

int  find_tree(int k)
{
    if(tree[k].l==tree[k].r)
    {
        return tree[k].sum;
    }
    int mid=(tree[k].l+tree[k].r)/2;
    if(k<=mid)
    {
        return find_tree(k*2);//左孩子
    }
    else
        return find_tree(k*2+1);//否则找右孩子
}

3、单点修改
原理=单点查询+建树的结合

void changdata(int k,int y)//k要查找的位置，y是要增加的值
{
    if(tree[k].l==tree[k].r)
    {
        tree[k].sum+=y;
        return ;
    }
    int m=(tree[k].l+tree[k].r)/2;
    if(x<=m) changdata(2*k,y);
    else changdata(2*k+1,y);
    
    tree[k].sum=tree[k*2].sum+tree[k*2+1].sum
}

4.区间查询

int ans=0;
void sum(int k,int x,int y)
{
    if(tree[k].l>=x && tree[k].r<=y)
    {
        ans+=tree[k].sum;
        return ;
    }
    int m=(tree[k].l+tree[k].r)/2;
    if(x<=m)    sum(2*k,x,y);
    if(y>m)       sum(2*k+1,x,y);
}

5、区间修改
引入懒惰标记

void down(int k)
{
    tree[k*2].f+=tree[k].f;
    tree[k*2+1].f+=tree[k].f;
    tree[k*2].w+=tree[k].f*(tree[k*2].r-tree[k*2].l+1);
    tree[k*2+1].w+=tree[k].f*(tree[k*2+1].r-tree[k*2+1].l+1);
    tree[k].f=0;
}

void add(int k)
{
    if(tree[k].l>=a&&tree[k].r<=b)//当前区间全部对要修改的区间有用 
    {
        tree[k].w+=(tree[k].r-tree[k].l+1)*x;//(r-1)+1区间点的总数
        tree[k].f+=x;
        return;
    }
    if(tree[k].f) down(k);//懒标记下传。只有不满足上面的if条件才执行，所以一定会用到当前节点的子节点 
    int m=(tree[k].l+tree[k].r)/2;
    if(a<=m) add(k*2);
    if(b>m) add(k*2+1);
    tree[k].w=tree[k*2].w+tree[k*2+1].w;//更改区间状态 
}