C语言 编写线段树

线段树同样也是一个比较冷门的数据结构,一般用于计算区域和,或者在修改某一个或几个节点的数据时,计算区域数据和。
首先线段树的特点是树的叶子全是数组中的数,而往上的节点都是相应子节点的和。
首先需要对数组创造相应的树结构。其中node为0也就是根节点,start和end指的是需要创建树的数组的起始和终点位置。而整个函数的思路为,mid为起点和终点的中点,根节点代表[start,end]的元素和,而它的两个子节点就是[start,mid]和[mid+1,end],同时接着往下走,把数组进一步进行划分,由于用的是递归的方法,所以需要考虑递归出口,而出口就是找到的区域中只含有一个数字也就是到了叶子,其中存放nums[start]的值即可,而往上的根节点就是两个叶子的和。
C语言 编写线段树
更新数组和树的思路是同样的,通过二分的方法去找到需要更改的叶子节点也即是start == end的时候,这时将nums[idx]和tree[node]改为相应的数值,同时将向上的根节点的值相应更改。
C语言 编写线段树
查询数组区域和,这地方也同样假如节点正好是所求区域就直接返回节点值,假如进入的区域在所求区域外直接返回0,假如找到叶子节点直接返回叶子节点的值。
C语言 编写线段树
用一个随手打的数组{1,2,11,25,31,5} 测试一下,计算[1,3]也就是2+11+25的结果:
C语言 编写线段树
结果如下:
C语言 编写线段树