线段树

转载

mob604756f44f2a 2021-09-08 16:39:00

文章标签 搜索区间修改线段树子树权值 文章分类 代码人生

线段树

第一部概念引入

线段树是一种二叉树，也就是对于一个线段，我们会用一个二叉树来表示。比如说一个长度为4的线段，我们可以表示成这样：

线段树_搜索

性质：节点i的权值=她的左儿子权值+她的右儿子权值。

我们知道，一颗二叉树，她的左儿子和右儿子编号分别是她*2和她*2+1
再根据刚才的性质，得到式子：tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;就可以建一颗线段树了！

1.建树：

代码如下：

//建树
void BuildTree(int left,int right,int i){
    tree[i].L=left;
    tree[i].R=right;

    if(left==right){//叶子节点
        return ;
    }
    int mid=(left+right)>>2;
    //递归构建左右子树
    BuildTree(left,mid,i*2);
    BuildTree(mid+1,right,i*2+1);
    tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
}

2.区间查询

线段树_子树_02

我们总结一下，线段树的查询方法：
1、如果这个区间被完全包括在目标区间里面，直接返回这个区间的值
2、如果这个区间的左儿子和目标区间有交集，那么搜索左儿子
3、如果这个区间的右儿子和目标区间有交集，那么搜索右儿子

代码：

//区间查询
int search(int left,int right,int i){
    //如果这个区间被完全包括在目标区间里面，直接返回这个区间的值
    if(tree[i].L>=left&&tree[i].R<=right){
        return tree[i].sum;
    }
    //如果这个区间和目标区间毫不相干，返回0
    if(tree[i].R<left||tree[i].L>right)return 0;
    int sum=0;
    //如果这个区间的左儿子和目标区间又交集，那么搜索左儿子
    if(tree[i*2].R>=left) sum+=search(left,right,i*2);
    //如果这个区间的右儿子和目标区间又交集，那么搜索右儿子
    if(tree[i*2+1].L<=right) sum+=search(left,right,i*2+1);
    return sum;
}

3.区间点修改

//区间点修改，区间点  dis  加 k 
void add(int i,int dis,int k){
    if(tree[i].L==tree[i].R){//如果是叶子节点，那么说明找到了
        tree[i].sum+=k;
        return ;
    }
    if(dis<=tree[i*2].R)  
        add(i*2,dis,k);//在左子树
    else  
        add(i*2+1,dis,k);//在右子树
    tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;//返回更新
    return ;
}

4.区间修改

区间修改和单点查询，我们的思路就变为：如果把这个区间加上k，相当于把这个区间涂上一个k的标记，然后单点查询的时候，
就从上跑道下，把沿路的标记加起来就好。这里面给区间贴标记的方式与上面的区间查找类似，原则还是那三条，只不过第一条：
如果这个区间被完全包括在目标区间里面，直接返回这个区间的值变为了如果这个区间如果这个区间被完全包括在目标区间里面，讲这个区间标记k。

// 区间修改
void add(int i,int left,int right,int k){
    if(tree[i].L>=left && tree[i].R<=right){//如果这个区间被完全包括在目标区间里面，将这个区间标记k
        tree[i].sum+=k;
        return ;
    }
    //如果这个区间的左儿子和目标区间又交集，那么搜索左儿子
    if(tree[i*2].R>=left)
        add(i*2,left,right,k);
    //如果这个区间的右儿子和目标区间又交集，那么搜索右儿子
    if(tree[i*2+1].L<=right)
        add(i*2+1,left,right,k);
}

6.单点查询

然后就是单点查询了，这个更好理解了，就是dis在哪往哪跑，把路径上所有的标价加上就好了：

//单点查询
int ans=0;
void search(int i,int dis){
    ans+=tree[i].sum;//一路加起来
    if(tree[i].L==tree[i].R)
        return ;
    if(dis<=tree[i*2].R)
        search(i*2,dis);
    if(dis>=tree[i*2+1].L)
        search(i*2+1,dis);
}