线段树详解以及C++实现代码

 更新时间:2021年07月18日 15:05:29   作者:唐 家 三 少  
线段树在一些acm题目中经常见到,这种数据结构主要应用在计算几何和地理信息系统中,这篇文章主要给大家介绍了关于线段树以及C++实现的相关资料,需要的朋友可以参考下

应用场景

假设有这样的问题:有n个数,m次操作,操作分为:修改某一个数或者查询一段区间的值

分析下,如果针对数组元素的修改可以是O(1)完成,求某个区间值需要O(n)才可以完成,如果m和n都很大的情况,这个复杂度就很难接受了。

我们之前学过的前缀和算法可以解决区间求和的问题,并且时间复杂度是O(1),但如果涉及到修改操作,前缀和数组都需要重新计算,时间复杂度也是O(n)

有没有什么办法可以兼顾以上两种操作,并且可以将时间复杂度降低?

这就是我们要学习的线段树!把修改和查询的时间复杂度都降到O(logn)!!!

算法思想

先来看下线段树长什么样:

有以下数组(为方便计算,数组下标从1开始)

我们把它转换成线段树,是长这样的:

1)叶子结点(绿色)存的都是原数组元素的值

2)每个父结点是它的两个子节点的值的和

3)每个父结点记录它表示区间的范围,如上图的“1-2”表示1到2的区间

下面我们来看看线段树是如何降低操作复杂度的!

查询操作

例如我们需要查询2-5区间的和

使用递归的思想:

2~5的和

=2~3的和+4~5的和

=3+5+4~5的和

=3+5+11

=19

总之,就是沿着线段树的划分把区间分开,再加到一块就行啦!

修改操作

例如,我们要把结点2的值由3->5,线段树需要沿着红色部分一个一个改,直到根结点:

不管是修改操作还是查询操作,时间复杂度都是O(logn)

下一步我们来看怎么实现线段树!

算法实现

首先我们需要将原始数组建立成一颗线段树,然后在树的基础上提供查询和修改的操作。

建树

观察上图,我们发现线段树是一棵近似完全二叉树,利用完全二叉树的性质,我们就可以直接用一个数组来存它。

就像上图一样把各个节点标上号,如果根节点编号是n,那它的左子树编号是2n,右子树的编号是2n+1

所以说,知道了根节点的编号,我们就可以快速有效的找到左右子树的根节点

void build(int root,int start,int end){
    if(start == end){
        tree[root] = num[start];
        return;
    }
    int leftroot = root * 2;//左结点
    int rightroot = root * 2 + 1;//右结点
    int mid = (start+end)/2;
    build(leftroot,start,mid);//递归计算左结点
    build(rightroot,mid+1,end);//递归计算右结点
    tree[root] = tree[leftroot] + tree[rightroot];//根结点值=左根+右根
}

查询

int query(int root,int start,int end,int l,int r){
    if(l<=start && r>= end){
        return tree[root];
    }
    int leftroot = root * 2;
    int rightroot = root * 2 + 1;
    int mid = (start+end)/2;
    int sum = 0;
    if(l<=mid){
        sum += query(leftroot,start,mid,l,r);
    }
    if(r>mid){
        sum += query(rightroot,mid+1,end,l,r);
    }
    return sum;
}

修改

/**
* 修改[l,r]区间里的数,都加上k值
* @param root
* @param start
* @param end
* @param l
* @param r
* @param k
*/
void update(int root,int start,int end,int l,int r,int k){
    if(start == end){
        tree[root] += k;
        return;
    }
    int leftroot = root * 2;
    int rightroot = root * 2 + 1;
    int mid = (start+end)/2;
    if(l<=mid){
        update(leftroot,start,mid,l,r,k);
    }
    if(r>mid){
        update(rightroot,mid+1,end,l,r,k);
    }
    tree[root] = tree[leftroot] + tree[rightroot];
}

!!!:考虑下按区间修改元素值的复杂度?

注意事项:

1)我们在实现线段树时,实际存储肯定大于原始数组,我们一般让tree数组的长度为原始数据长度的3-4倍。

2)本文只是为了让大家学习线段树的实现原理,实际中我们可以将原始数组的start,end使用结构体存储,这样更简洁

总结

到此这篇关于线段树详解以及C++实现的文章就介绍到这了,更多相关C++实现线段树内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • C语言约瑟夫环的实现

    C语言约瑟夫环的实现

    这篇文章主要介绍了C语言约瑟夫环的实现的相关资料,这里主要是利用数据数据结果中循环链表来实现,需要的朋友可以参考下
    2017-08-08
  • C++ 使用getline()从文件中读取一行字符串方法示例

    C++ 使用getline()从文件中读取一行字符串方法示例

    这篇文章主要介绍了C++ 使用getline()从文件中读取一行字符串方法示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-09-09
  • STL  priority_queue(优先队列)详解

    STL priority_queue(优先队列)详解

    这篇文章主要介绍了 STL priority_queue(优先队列)详解的相关资料,需要的朋友可以参考下
    2016-10-10
  • C语言连续生成随机数的实现方法

    C语言连续生成随机数的实现方法

    这篇文章主要介绍了C语言连续生成随机数的实现方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2021-01-01
  • C语言中进行大小写字母转化的示例代码

    C语言中进行大小写字母转化的示例代码

    C语言标准库中提供了用于大小写转换的函数,使得这一操作变得简单而高效,本文将详细介绍如何在C语言中进行大小写字母的转换,包括相关的函数和示例代码,需要的朋友可以参考下
    2024-03-03
  • C++求阶乘的两种方法

    C++求阶乘的两种方法

    这篇文章主要介绍了C++求阶乘的两种方法,有需要的朋友可以参考一下
    2013-12-12
  • c语言中比较特殊的输入函数举例详解

    c语言中比较特殊的输入函数举例详解

    C语言提供了丰富的标准库函数,用于处理各种输入输出操作,下面这篇文章主要介绍了c语言中比较特殊的输入函数的相关资料,文中通过代码介绍的非常详细,需要的朋友可以参考下
    2024-09-09
  • c++访问私有private成员变量的常用方法

    c++访问私有private成员变量的常用方法

    这篇文章主要介绍了c++访问私有private成员变量的常用方法,需要的朋友可以参考下
    2020-08-08
  • C++ 动态内存分配详解(new/new[]和delete/delete[])

    C++ 动态内存分配详解(new/new[]和delete/delete[])

    这篇文章主要介绍了C++ 动态内存分配详解(new/new[]和delete/delete[]),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2021-05-05
  • 学好C++必须做到的50条 绝对经典!

    学好C++必须做到的50条 绝对经典!

    学好C++必须做到的50条,绝对经典!想要学好C++的朋友一定要认真阅读本文,更要做到以下50条
    2016-09-09

最新评论