hdu_1166敌兵布阵

在网上下到acm大牛总结的线段树专辑，其代码风格我学到了很多。

线段树，类似区间树，是完全二叉树，它在各个节点保存一条线段（数组中的一段子数组），主要用于高效解决连续区间的动态查询问题，由于二叉结构的特性，它基本能保持每个操作的复杂度为O(lgN)!

性质：父亲的区间是[a,b],用二分法的思想去分左右儿子，左儿子为[a,(a+b)/2]，右儿子为[(a+b)/2+1,b],因此线段树所需空间为最大数的4倍

首先宏定义左右儿子：

#define lson l,m,rt<<1<span>	//结点为rt,区间为[l,m]</span>
#define rson m+1,r,rt<<1|1 <span>	//结点为rt,区间为[m+1,r]</span>

计算父节点的总值，总值为左右儿子值的总和：

<span style="font-family:FangSong_GB2312;">inline void fatherSum(int rt){
	sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}</span>

用递归思想构造线段树：

<span style="font-family:FangSong_GB2312;">void buildTree(int l,int r,int rt){
	if(l == r){
		scanf("%d",&sum[rt]);
		return;
	}
	int m = (l+r)>>1;
	buildTree(lson);
	buildTree(rson);
	fatherSum(rt);
} </span>

线段树对应结点更新以及父节点更新以及父节点的父节点的更新...，树根遍历到对应结点所经过的结点的值都要更新：

<span style="font-family:FangSong_GB2312;">void update(int p,int num,int l,int r,int rt){
	if(l == r){
		sum[rt] += num;
		return;
	}
	int m = (l+r)>>1;
	if(p <= m)
		update(p,num,lson);
	else
		update(p,num,rson);
	fatherSum(rt);
}</span>

区间查询：

<span style="font-family:FangSong_GB2312;">int query(int L,int R,int l,int r,int rt){
	if(L <= l && r <= R)
		return sum[rt];
	int m = (l+r)>>1;
	int ret = 0;
	if(L <= m) 	
		ret += query(L,R,lson);
	if(R > m)	
		ret += query(L,R,rson);
	return ret;
}</span>