线段树

线段树

线段树是一种二叉搜索树，与区间树相似，每一个叶子代表一个区间，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。

使用线段树可以快速的查找某一个节点在若干条线段中出现的次数，时间复杂度为O(logN）。而未优化的空间复杂度为2N，因此有时需要离散化让空间压缩。

如图所示，每一个节点都可以代表一个区间，子节点则分别表示父节点的左右半区间，例如父亲的区间是[a,b]，那么(c=(a+b)/2)左儿子的区间是[a,c]，右儿子的区间是[c+1,b]。下面看几种线段树的基本操作：

1.创建线段树

我们先定义一个用来储存数据的公用结构体：(题目中的要求的结果不同，结构体也不同，当要求求各个节点的和时，自然不用定义最大值与最小值)

struct Node
{
	int l;//储存左边界 // 
	int r;//储存右边界 //
	int sum;//储存各个节点的和 //
	int Max;//储存节点中最大数据 //
	int Min;//储存节点中最小数据//
}Tree[1000]//数组的大小根据提议而定//;

下面我们将构建线段树：

void Build(int o,int l,int r)
{
	//首先记录l和r的值 
	Tree[o].l = l;
	Tree[o].r = r;
	if (l == r)		//到达最底层，递归终止,就是上面那个图，最底层左边界与右边界是相等的，用这个判断是否到达最底层 
	{
		int t;
		scanf ("%d",&t);		//输入最底层数据，因为只有最底层的数据时输入的，其他的都是根据最底层推出来的 
		Tree[o].sum = Tree[o].Max = Tree[o].Min = t;		//更新节点数据 
		return;
	}
	int mid = (l+r) >> 1;		//找到中间节点 
	Build(o*2 , l , mid);		//递归建左子树 
	Build(o*2+1 , mid+1 , r);		//递归建右子树 
	PushUp(o);		//更新当前节点的值 
}

下面解释一下最后一步PushUp(o)更新节点值(注意不是节点更新)的函数：

void PushUp(int o)
{
	Tree[o].sum = Tree[o*2].sum + Tree[o*2+1].sum;
	Tree[o].Max = max(Tree[o*2].Max,Tree[o*2+1].Max);
	Tree[o].Min = min(Tree[o*2].Min,Tree[o*2+1].Min);
}(max,min时c++中两个常用的函数，头文件时#include<algorithm>加上using namespace std)

同过以上的过程便成功构建了线段树。

2.查询任意区间的最大值，最小值，区间和等题目要求输出的结果：

我们以查询x到y区间和为例：

int QuerySum(int o,int l,int r,int x,int y)		//查找x到y的和 
{
	if (l == x && r == y)		//如果恰好是当前节点，就返回 
	{
		return Tree[o].sum;
	}
	int mid = ( l + r ) / 2;
	if (mid >= y)		//全在左边 
		return QuerySum(o*2,l,mid,x,y);
	else if (x > mid)		//全在右边 
		return QuerySum(o*2+1,mid+1,r,x,y);
	else		//一半在左一半在右 
		return QuerySum(o*2,l,mid,x,mid) + QuerySum(o*2+1,mid+1,r,mid+1,y);
} //利用递归的方式一步步的查询//

3.将x借点更新为y节点：

void UpDate(int o,int l,int r,int x,int y)		//把x节点更新为y
{
	if (l == r)		//递归结束
	{
		Tree[o].Max = Tree[o].Min = Tree[o].sum = y;		//精确找到了节点，更新 
		return;
	}
	int mid = (l+r) / 2;		//找到中间位置
	if (x <= mid)
		UpDate(o*2,l,mid,x,y);		//找左子树 
	else
		UpDate(o*2+1,mid+1,r,x,y);		//找右子树 
	PushUp(o);		//更新当前节点 
}//利用递归一步步找到x节点//

下面看一个例题：

士兵杀敌（一）

时间限制：1000 ms  |  内存限制：65535 KB

难度：3

描述

南将军手下有N个士兵，分别编号1到N，这些士兵的杀敌数都是已知的。

小工是南将军手下的军师，南将军现在想知道第m号到第n号士兵的总杀敌数，请你帮助小工来回答南将军吧。

注意，南将军可能会问很多次问题。

输入

只有一组测试数据
第一行是两个整数N,M，其中N表示士兵的个数(1<N<1000000)，M表示南将军询问的次数(1<M<100000)
随后的一行是N个整数，ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行有两个整数m,n，表示南将军想知道第m号到第n号士兵的总杀敌数（1<=m,n<=N)。

输出

对于每一个询问，输出总杀敌数
每个输出占一行

样例输入

5 2
1 2 3 4 5
1 3
2 4

样例输出

6
9

先看一下该题利用上述函数方式，解决代码：

#include<cstdio>
#include<algorithm>
using namespace std;
#define L o<<1
#define R (o<<1)|1
struct Node
{
	int l;//储存左边界 // 
	int r;//储存右边界 //
	int sum;//储存各个节点的和 //
}Tree[1000<<2];
void PushUp(int o)
{
	Tree[o].sum = Tree[o*2].sum + Tree[o*2+1].sum;
}
void Build(int o,int l,int r)
{
	//首先记录l和r的值 
	Tree[o].l = l;
	Tree[o].r = r;
	if (l == r)		//到达最底层，递归终止,就是上面那个图，最底层左边界与右边界是相等的，用这个判断是否到达最底层 
	{
		int t;
		scanf ("%d",&t);		//输入最底层数据，因为只有最底层的数据时输入的，其他的都是根据最底层推出来的 
		Tree[o].sum = t;		//更新节点数据 
		return;
	}
	int mid = (l+r) >> 1;		//找到中间节点 
	Build(o*2 , l , mid);		//递归建左子树 
	Build(o*2+1 , mid+1 , r);		//递归建右子树 
	PushUp(o);		//更新当前节点的值 
}
int QuerySum(int o,int l,int r,int x,int y)		//查找x到y的和 
{
	if (l == x && r == y)		//如果恰好是当前节点，就返回 
	{
		return Tree[o].sum;
	}
	int mid = (l + r) / 2;
	if (mid >= y)		//全在左边 
		return QuerySum(o*2,l,mid,x,y);
	else if (x > mid)		//全在右边 
		return QuerySum(o*2+1,mid+1,r,x,y);
	else		//一半在左一半在右 
		return QuerySum(o*2,l,mid,x,mid) + QuerySum(o*2+1,mid+1,r,mid+1,y);
} 
int main()
{
	int n,m;
	scanf ("%d%d",&n,&m);
	Build(1,1,n);
	int x,y;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		printf ("%d\n",QuerySum(1,1,n,x,y));
	}
	
	return 0;
}

该题还有一种简单的方法，就是利用数组a[i]记录前i组数据的和，当给出区间x，y时用sum[y] - sum[x-1]即可，代码如下：

#include <stdio.h>  
int a[1123456];  
int sum[1123456];  //由于mian函数的储存空间有限，当数组比较大时，通常将数组定义在main函数的前面//
int main()  
{  
    int n,m;   
    int x,y; 
    scanf("%d %d", &n, &m);   
    for(int i = 1;i <= n;i++){  
        scanf("%d",&a[i]);  
        sum[i] = sum[i-1]+a[i];//sum[i]表示前i个数的总和//  
    }  
 
    for(int i = 0;i < m;i++)
	{  
        scanf("%d %d", &x, &y);  
        printf("%d\n", sum[y]-sum[x-1]);
    }  
  	return 0;  
}  

练习网站：https://cn.vjudge.net/contest/179845 

愿你一生清澈明朗，做你愿做之事，爱你愿爱之人！