线段树的入门级

    线段树是一种二叉搜索树,与区间树相似,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
      对于线段树中的每一个非叶子节点[a,b],它的左儿子表示的区间为[a,(a+b)/2],右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树,最后的子节点数目为N,即整个线段区间的长度。
      使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为O(logN)。而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩。
----来自百度百科

【以下以 求区间最大值为例】
先看声明:

#include <stdio.h>
#include <math.h>
const int MAXNODE = 2097152;
const int MAX = 1000003;
struct NODE{
    int value;        // 结点对应区间的权值
    int left,right;   // 区间 [left,right]
}node[MAXNODE];
int father[MAX];     // 每个点(当区间长度为0时,对应一个点)对应的结构体数组下标



【创建线段树(初始化)】:

       由于线段树是用二叉树结构储存的,而且是近乎完全二叉树的,所以在这里我使用了数组来代替链表上图中区间上面的红色数字表示了结构体数组中对应的下标。

在完全二叉树中假如一个结点的序号(数组下标)为 I ,那么 (二叉树基本关系)

I 的父亲为 I/2,

I 的另一个兄弟为 I/2*2 或 I/2*2+1

I 的两个孩子为 I*2 (左)   I*2+1(右)

有了这样的关系之后,我们便能很方便的写出创建线段树的代码了。

void BuildTree(int i,int left,int right){ // 为区间[left,right]建立一个以i为祖先的线段树,i为数组下标,我称作结点序号
	node[i].left = left;	// 写入第i个结点中的 左区间
	node[i].right = right;	// 写入第i个结点中的 右区间
	node[i].value = 0; 		// 每个区间初始化为 0
	if (left == right){	// 当区间长度为 0 时,结束递归
		father[left] = i; // 能知道某个点对应的序号,为了更新的时候从下往上一直到顶
		return;
	}
	// 该结点往 左孩子的方向 继续建立线段树,线段的划分是二分思想,如果写过二分查找的话这里很容易接受
	// 这里将 区间[left,right] 一分为二了
	BuildTree(i<<1, left, (int)floor( (right+left) / 2.0));
	// 该结点往 右孩子的方向 继续建立线段树
	BuildTree((i<<1) + 1, (int)floor( (right+left) / 2.0) + 1, right);
}


 

【单点更新线段树】:

       由于我事先用 father[ ] 数组保存过 每单个结点 对应的下标了,因此我只需要知道第几个点,就能知道这个点在结构体中的位置(即下标)了,这样的话,根据之前已知的基本关系,就只需要直接一路更新上去即可。

 

<pre class="html" name="code">void UpdataTree(int ri){ // 从下往上更新(注:这个点本身已经在函数外更新过了)

	if (ri == 1)return;	// 向上已经找到了祖先(整个线段树的祖先结点 对应的下标为1)
	int fi = ri / 2; 		// ri 的父结点
	int a = node[fi<<1].value; // 该父结点的两个孩子结点(左)
	int b = node[(fi<<1)+1].value; // 右
	node[fi].value = (a > b)?(a):(b);	// 更新这个父结点(从两个孩子结点中挑个大的)
	UpdataTree(ri/2);		// 递归更新,由父结点往上找
}

 


【查询区间最大值】:
       将一段区间按照建立的线段树从上往下一直拆开,直到存在有完全重合的区间停止。对照图例建立的树,假如查询区间为 [2,5]

红色的区间为完全重合的区间,因为在这个具体问题中我们只需要比较这 三个区间的值 找出 最大值 即可。

int Max = -1<<20;
void Query(int i,int l,int r){ // i为区间的序号(对应的区间是最大范围的那个区间,也是第一个图最顶端的区间,一般初始是 1 啦)
	if (node[i].left == l && node[i].right == r){ // 找到了一个完全重合的区间
		Max = (Max < node[i].value)?node[i].value:(Max);
		return ;
	}
	i = i << 1; // get the left child of the tree node
	if (l <= node[i].right){ // 左区间有涉及
		if (r <= node[i].right) // 全包含于左区间,则查询区间形态不变
			Query(i, l, r);
		else // 半包含于左区间,则查询区间拆分,左端点不变,右端点变为左孩子的右区间端点
			Query(i, l, node[i].right);
	}
	i += 1; // right child of the tree
	if (r >= node[i].left){ // 右区间有涉及
		if (l >= node[i].left) // 全包含于右区间,则查询区间形态不变
			Query(i, l, r);
		else // 半包含于左区间,则查询区间拆分,与上同理
			Query(i, node[i].left, r);
	}
}


HDU  1754解题报告

// HDU 1754  I Hate It  线段树 -- 单点更新
// 有效结点: 20w
// 深度达到:(lg20w)/(lg2) +1 .= 18.61 .= 19
// 其完全二叉树 总结点 个数为: (1<<19) - 1 个
// 区间最大值问题
//
/*test data
5 6
1 2 3 4 5
Q 1 5
U 3 6
Q 3 4
Q 4 5
U 2 9
Q 1 5

*/
#include <cstdio>
#include <cmath>
#include <iostream>
using namespace std;

const int MAXNODE = 524288; // 1<<19
const int MAXST = 200001;
struct STU{
	int grade;
	int left,right;
}st[MAXNODE];

int father[MAXST];

void BuildTree(int i,int left,int right){ // i是结点的序号 对应了数组下标
	st[i].left = left;
	st[i].right = right;
	st[i].grade = 0; // 初始化为0
	if (left == right){
		father[left] = i; // 为了更新的时候从下往上 一直到顶
		return;
	}
	BuildTree(i<<1, left, (int)floor( (right+left) / 2.0));
	BuildTree((i<<1) + 1, (int)floor( (right+left) / 2.0) + 1, right);
}

void UpdataTree(int ri){ // 从下往上更新

	if (ri == 1){return;}
	int fi = ri / 2; // 父结点
	int a = st[fi<<1].grade; // 该父结点的两个子结点
	int b = st[(fi<<1)+1].grade;
	st[fi].grade = (a > b)?(a):(b);
	UpdataTree(ri/2);

}

int Max;
void Query(int i,int l,int r){ // i为区间的序号,四段查询 即四种情况

	if (st[i].left == l && st[i].right == r){ // 找到了一个完全重合的区间
		Max = (Max < st[i].grade)?st[i].grade:(Max);
		return ;
	}
	i = i << 1; // left child of the tree
	if (l <= st[i].right){ // 左区间有覆盖
		if (r <= st[i].right) // 全包含于左区间
			Query(i, l, r);
		else // 半包含于左区间
			Query(i, l, st[i].right);
	}
	i += 1; // right child of the tree
	if (r >= st[i].left){ // 右区间有覆盖
		if (l >= st[i].left) // 全包含于右区间
			Query(i, l, r);
		else // 半包含于左区间
			Query(i, st[i].left, r);
	}
}

int main()
{
//	freopen("in.txt","r",stdin);
	int n_s,n_q,igrade; 
	while(scanf("%d %d",&n_s,&n_q) != EOF){
		BuildTree(1, 1, n_s); 
		for (int i= 1 ; i <= n_s; i++){
			scanf("%d",&igrade);
			st[father[i]].grade = igrade; // 底层的无条件更新成绩
			UpdataTree(father[i]);
		}
		while(n_q--){
			char o[3];int a,b;
			scanf("%s %d %d",o,&a,&b);
			if ( o[0] == 'Q'){
				Max = 0;
				Query(1, a, b);
				printf("%d\n",Max);
			}else{
				st[father[a]].grade = b; // 底层的无条件更新成绩
				UpdataTree(father[a]);
			}
		}
	}
	return 0;
}

总的来说,这篇博客对线段树的基本操作讲的非常清晰,适合新手,但代码量有点大,在下一篇给出仿照not only success这位大神写的线段树模板


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值