线段树在算法中的应用[朱凯迪]

线段树在算法中的应用

作者：朱凯迪

作者单位：宁波工程学院

Email:zukaidi@https://www.360docs.net/doc/ce7628477.html,

摘要：计算机信息学竞赛中出现了越来越多的统计，查找，规划，排序，染色等等的题目。平衡二叉树和线段树是两种最常见的解决此类问题的数据结构。可是平衡二

叉树有一个缺点，就是变成复杂度很高。我们可以看到在某些题目中，线段树是

它的有力替代品。这篇论文主要介绍了线段树的操作，优化以及应用。该论文也

会系统地介绍染色问题使用线段树的一般解法。

关键词：线段树数据结构信息学算法

1.线段树的定义及特征

一棵二叉树，记为T(a, b)，参数a, b表示该结点表示的区间[a, b]。区间长度b-a记为L。递归定义T[a, b]：

若L>1：[a, (a + b) div 2]为T的左儿子

[(a + b) div 2, b]为T的右儿子。

若L=1：T为一个叶子结点。

表示区间[1, 10]的线段树表示如图1-1所示：

- 1 -

- 2 -

定理1：线段树把区间任意一条线段都分成不超过L log 2条线段。 证明：

① 在区间(a, b)中，对于线段(c, d)，如果(c <= a)或(d >= b)，那么线段在(a, b)中被

分为不超过)log(a b -。

用归纳法证明，如果是单位区间，最多被分为一段，成立。 如果区间(a, b)的左儿子与右儿子成立，那么如果当c <= a 时，

1． 若d <= (a + b) div 2那么相当与其左儿子分该线段，所分该线段树不超过

)2)log((a div b a -+，即不超过)log(a b -，成立。

2． 若d > (a + b) div 2那么相当于该线段被分为它左儿子表示的线段，加上右

儿子分该线段，线段数不超过)2)(log(1div b a b +-+，也不超过

)log(a b -，成立。

对于d >= b 的情况证明类似，不在赘述。

② 在区间(a, b)中，对于任意线段也用归纳法证明。

对于单位区间，最多分为一段，成立。

若(a, b)的左儿子与右儿子均成立，则对于线段(c, d)

1． 若d <= (a +b) div 2则该区间所分该线段等于其左儿子区间所分该线

段，线段数小于)log(2)2)

(log(a b a div b a b -<-+-，成立。 2． 若c > (a + b) div 2则该区间所分该线段等于其右儿子区间所分该线

段，线段数小于)log(2)2)

(log(a b div b a b -<+-，成立。 3． 若1、2均不成立，则此线段在左儿子区间分该线段满足d > V.Lson.b 1，分

该线段数不超过)2)

(log(div b a b +-，而在右儿子区间分该线段满足c <= V.Rson.a 2，分该线段不超过)12)

(log(-+-div b a b ，所以在该区间分该线段不超过)log(2a b -，成立。

这个结论为线段数能在)(log L O 的时间内完成一条线段的插入、删除、查找等工作提供了理论依据。

除了以上性质，线段数还具有以下一些性质： ● 线段数是一个平衡树，树的高度为N log 。

● 任两个结点要么是包含关系要么没有公共部分，不可能重叠。

● 给定一个叶子p ，从根到p 路径上所有结点代表的区间都包含点p ，且其它结点代

表的区间都不包含p 。

2. 线段树的基本存储结构和操作

1 Lson 为Left Son 的缩写，表示左儿子。 2

Rson 为Reft Son 的缩写，表示右儿子。

- 3 -

2.1 线段数的基本存储结构

线段数的一个结点的最基本存储数据结构如图2-1-1所示：

(图2-1-1)

也可以用数组模拟二叉树，则结构体中不需要两个指针变量。

其中left 和right 分别表示该结点的左右端点，而mid

则是中点。这样就不需要在每次再计算了。而lson 和rson 分别指向该结点的左儿子和右儿子，如果没有，则为NULL 。

这只是线段树结点的最基本结构，在解决实际问题时，还需要根据实际情况添加各种需要储存的数据。如[ZOJ]1610 Count the Colors 3中，我建立的线段树结点结构体如图2-1-2所示：

(图2-1-2)

其中l, r 各代表左右端点，mid 代表中点，col 代表颜色。lc 和rc 各代表左儿子和右儿子。

2.2 线段数的基本操作

2.2.1 线段树的建立操作

在对线段树进行操作前，我们需要建立起线段树的结构。我们使用结构 体数组来保存线段树，这样对于非叶节点，若它在数组中编号为 num ，则其 左右子节点的编号为 2 * num ，2 * num + 1。由于线段树是二分的树型结

构，我们可以用递归的方法，从根节点开始来建立一棵线段树。代码如图2-2-1所 示：

3

https://www.360docs.net/doc/ce7628477.html,/onlinejudge/showProblem.do?problemCode=1610，染色问题。下文会具体讲解。

(图2-2-1)

对应不同的题目，我们会在线段树节点中添加另外的数据域，并随着线段的插入或删除进行维护，要注意在建树过程中将这些数据域初始化。

2.2.2 线段树的插入操作

为了在插入线段后，我们能知道哪些节点上的线段被插入（覆盖）过。我们需要在节点中添加一个cover域，来记录当前节点所表示的线段是否被覆盖。这样，在建树过程中，我们需要把每个节点的cover 域置0；

在线段的插入过程中，我们从根节点开始插入，同样采取递归的方法。如果插入的线段完全覆盖了当前节点所代表的线段，则将当前节点的cover 域置 1 并返回。否则，将线段递归进入当前节点的左右子节点进行插入。代码图2-2-2所示。

- 4 -

(图2-2-2)

要注意，这样插入线段时，有可能出现以下这种情况，即先插入线段[1,3)，再插入线段[1,5)。这样，代表线段[1,3)的节点以及代表线段[1,5)的节点的cover 值均为1，但是在统计时，遇到这种情况，我们可以只统计更靠近根节点的节点，因为这个节点所代表的线段包含了其子树上所有节点所代表的线段。

2.2.3 线段树的删除操作

线段树的删除操作跟插入操作不大相同，因为一条线段只有被插入过才能被删除。比如插入一条线段[3,10)，则只能删除线段[4,6)，不能删除线段[7,12)。当删除未插入的线段时，操作返回false 值。

我们一样采用递归的方法对线段进行删除，如果当前节点所代表的线段未被覆盖，即cover 值为0，则递归进入此节点的左右子节点进行删除。而如果当前节点所代表的线段已被覆盖，即cover 值为1，则要考虑两种情况。一是删除的线段完全覆盖当前节点所代表的线段，则将当前节点的cover值置0。我们应该递归的在当前节点的子树上所有节点删除线段。另一种情况是删除的线段未完全覆盖当前节点所代表的线段，比如当前节点代表的线段为[1,10)，而要删除的线段为[4,7)，则删除后剩下线段[1,4)和[7,10)，我们采用的方法是，将当前节点的cover置0，并将其左右子节点的cover置1，然后递归的进入左右子节点进行删除。删除操作的代码如图2-2-3所示：

- 5 -

图2-2-3

相对插入操作，删除操作比较复杂，需要考虑的情况很多，稍有不慎就会出错，在比赛中写删除操作时务必联系插入操作的实现过程，仔细思考，才能避免错误。

2.2.4 线段树的统计操作

对应不同的问题，线段树会统计不同的数据，比如线段覆盖的长度，线段覆盖连续区间的个数等等。其实现思路不尽相同，我们以下以统计线段覆盖长度为例，简要介绍线段树统计信息的过程。文章之后的章节会讲解一些要用到线段树的题目，并会详细介绍线段树的用法，以及各种信息的统计过程。

对于统计线段覆盖长度的问题，可以采用以下的思路来统计信息，即从根节点开始搜索整棵线段树，如果当前节点所代表的线段已被覆盖，则将统计长度加上当前线段长度。否则，递归进入当前节点的左右子节点进行统计。实现代码图2-2-4所示：

- 6 -

(图2-2-4)

小结：线段树作为一种数据结构只是解决问题的一个工具，具体的使用方法则非常灵活。以上介绍的仅仅为线段树的基础，在实际应用中，需要针对待解的题目设计节点存储信息，以及修改维护操作等等。下面将由浅及深的介绍线段树的一些应用，其中的一些线段树使用方法值得思考和借鉴。

3.线段树的应用举例

3.1 染色问题

问题链接：https://www.360docs.net/doc/ce7628477.html,/onlinejudge/showProblem.do?problemCode=1610

代码链接：https://www.360docs.net/doc/ce7628477.html,/code/u/XadillaX/ZJU/16104

问题大意：输入n个有色线段[a,b]，问最后每种颜色有多少连续段？（所有数字在[0,8000]范围内）

解题思路：这里是一个结构体：

l、r各代表左右端点，mid代表中点，col代表颜色。lc和rc各代表左儿子和右儿子。

1.创建线段树

1)取最小和最大的两个数作为端点，建立线段树。

2)当前节点的两个端点值之差等于一时，此时该节点即位叶子节点，不

用再向下分。

3)否则，分裂该节点为[a，(a + b) / 2], [(a + b) / 2, b];

4)创建线段树时，注意初始化操作。

4ACMDIY平台为作者自主开发的一个在线代码库平台。

- 7 -

2.线段树着色（根据不同的题目此操作各不相同，对zju_1610做分析）

1)当前节点的颜色与将要涂的颜色color相同，直接return。

2)当前线段树节点的两个端点和要涂的两个端点正好都相同，则将该节

点着为color，然后return。

3)要涂的两个端点在当前节点的两个端点之间时：先将当前节点的颜色

向其子节点扩展，然后：

①要涂的右端点小于或等于当前节点middle = (a + b) / 2时，向左子

节点移动。

②要涂的左端点大于或等于当前节点middle = (a + b) / 2时，向右子

节点移动。

③else （1 ，2）向左右子节点移动。

源代码：

/*

* File: p1610.cpp

* Author: xadillax

*

* Created on May 4, 2010, 7:20 PM

*/

#include

#include

#include

#include

#define NOCOL -1

#define MULCOL -2

using namespace std;

struct tree{

int l, r, col, mid;

tree *lc, *rc;

};

tree *root;

int n, l, r, col, i;

int color[8001], cnt[8001];

tree *init(int l, int r) //建立一棵线段树

{

tree *rst;

rst = (tree *)malloc(sizeof(tree));

rst->l = l, rst->r = r, rst->mid = (l + r) / 2, rst->col = NOCOL;

if(r - l == 1)

{

rst->lc = NULL;

rst->rc = NULL;

}

else

{

rst->lc = init(rst->l, rst->mid);

rst->rc = init(rst->mid, rst->r);

}

- 8 -

return rst;

}

void put(tree *p, int l, int r, int col)

{

if(p->l == l && p->r == r)

{

p->col = col;

return;

}

if(p->col != MULCOL)

{

p->lc->col = p->col;

p->rc->col = p->col;

p->col = MULCOL;

}

if(l < p->mid && r > p->mid)

{

put(p->lc, l, p->mid, col);

put(p->rc, p->mid, r, col);

}

else

{

if(p->mid <= l) put(p->rc, l, r, col);

else put(p->lc, l, r, col);

}

}

void cal(tree *p, int *color)

{

int i;

if(p->col == MULCOL)

{

cal(p->lc, color);

cal(p->rc, color);

}

else

if(p->col != NOCOL)

for(i = p->l; i < p->r; i++) color[i] = p->col; else

for(i = p->l; i < p->r; i++) color[i] = NOCOL; free(p);

}

int main()

{

while((scanf("%d", &n)) != EOF)

{

root = init(0, 8000);

memset(color, 0, sizeof(color));

memset(cnt, 0, sizeof(cnt));

while(n--)

{

scanf("%d%d%d", &l, &r, &col);

put(root, l, r, col);

}

cal(root, color);

if(color[0] != NOCOL) cnt[color[0]]++;

- 9 -

for(i = 1; i <= 7999; i++)

if(color[i] != NOCOL && color[i] != color[i - 1])

{

cnt[color[i]]++;

}

for(i = 0; i <= 8000; i++)

if(cnt[i] > 0) printf("%d %d\n", i, cnt[i]);

printf("\n");

}

return 0;

}

3.2 城市景观

问题链接：https://www.360docs.net/doc/ce7628477.html,/JudgeOnline/problem?id=3277

问题大意：如图所示，在一条水平线上有N个建筑物，建筑物都是长方形的，且可以互相遮盖。给出每个建筑物的左右坐标值Ai,Bi 以及每个建筑物的高度Hi，

需要计算出这些建筑物总共覆盖的面积。

题目数据范围：

建筑物个数N：1 <= N <= 40000

建筑物左右坐标值Ai, Bi：1 <= Ai,Bi <= 109

建筑物的高度Hi：1 <= Hi <= 109

解题思路：因为区间最大到10的9次方，开这么大的空间内存肯定不够，所以要离散化，用map存入然后用iterator遍历得到的有序序列存入vector。然后以vector

的下标建立线段树，统计时若结点不是叶子结点，则它的值为左右孩子的值

之和，否则返回底*高。

源代码：5

#include

#include

#include

using namespace std;

typedef struct nod

{

int left,right,mid;

long long h;

bool cover;

}node,*nd;

typedef struct p

{

long long start,end,hi;

}point;

// 元素个数上限和存储数组

const int MAX_NUM=200000; //第一次交re,干脆开个大的

node seg_tree[3*MAX_NUM+1];

point list[40005];

5此源代码来自文献参考中的第七项。

- 10 -

map my_map;

map::iterator it;

vector vec;

// 创建, l为左端点，r为右端点，num为在数组中的编号

void make(int l,int r,int num)

{

seg_tree[num].left=l;

seg_tree[num].right=r;

seg_tree[num].h=0;

seg_tree[num].mid= (l+r)/2;

if( l+1 != r )

{

make(l,seg_tree[num].mid,2*num);

make(seg_tree[num].mid,r,2*num+1);

}

}

// 插入操作，参数意义同上

void Insert(int l,int r,int num ,long long &hi)

{

if( seg_tree[num].left==l && seg_tree[num].right==r )

{

if(seg_tree[num].h

return ;

}

if( r<=seg_tree[num].mid )

Insert(l,r,2*num,hi);

else if( l>=seg_tree[num].mid )

Insert(l,r,2*num+1,hi);

else

{

Insert(l,seg_tree[num].mid,2*num,hi);

Insert(seg_tree[num].mid,r,2*num+1,hi);

}

}

//统计操作

long long cal(long long hi,int num)

{

if(hi>seg_tree[num].h) seg_tree[num].h=hi;

if( seg_tree[num].left+1 == seg_tree[num].right )

{

return seg_tree[num].h*( vec[ seg_tree[num].right ] - vec[ seg_tree[num].left ] ); }

return cal( seg_tree[num].h , 2*num ) + cal( seg_tree[num].h , 2*num+1 );

}

int main()

{

int T,t=1;

cin>>T;

long long a,b,h;

for(int i=0;i

{

cin>>a>>b>>h;

my_map[a]=0;

my_map[b]=0;

list[i].start=a; list[i].end=b; list[i].hi=h;

}

vec.push_back(0); //压入一个元素使vec从下标1开始有效

for( it=my_map.begin(); it!=my_map.end();++it )

{

- 11 -

it->second=t++;

vec.push_back( it->first );

}

vec.push_back( vec[vec.size()-1] +1 ); //将离散化之后的元素按顺序压入向量

make(1,my_map.size()+1,1);

for(int i=0;i

Insert( my_map[ list[i].start ] , my_map[ list[i].end ] ,1 , list[i].hi );

cout<

return 0;

}

4.线段树的应用总结

线段树是一种高效的数据结构。它的思想就是分治，非常基本也非常强大。在学习线段树的过程中，要时刻记住，线段树只是一种解题的工具，学习线段树只是学习一种解题的思维，就像图论中的BFS6一样。至于在具体的题目中如何去运用这个工具，则灵活的去建树并维护相关信息。这也需要在平时进行积累，做题时才能应用熟练。

5.线段树的练习题目推荐

以下题目是各OJ 上比较有名的线段树题目

Whu OnlineJudge

https://www.360docs.net/doc/ce7628477.html,/oak

题目号：1071 1224 1361 1344

Pku OnlineJudge

https://www.360docs.net/doc/ce7628477.html,/JudgeOnline

题目号3225 2482 1177 1029 2182 2750 2104 2528 2828 2777 2886 2761

Zju OnlineJudge

https://www.360docs.net/doc/ce7628477.html,

题目号2301 1128 1659 2112

参考文献

[1]岳云涛.浅谈线段数在信息学竞赛中的应用[J].

[2]林涛.线段数的应用[J],2004.

[3]刘汝佳, 黄亮.算法艺术与信息学竞赛[M]. 北京：清华大学出版社，2004.

[4]Thomas H.Cormen, Charles E.Leiserson, Ronald L.Rivest et al. Introduction to Algorithms[M]. 2nd Edition. America: The MIT Press, 2001.

[5]廖劲宇. pku 3277 (线段树+离散化)[OL]. https://www.360docs.net/doc/ce7628477.html,/liaojinyu282/archive/2010/01/16/5200521.aspx, 2010.

[6]朱凯迪. [ZOJ]1610 Count the Colors[OL]. https://www.360docs.net/doc/ce7628477.html,/kdtalker/blog/item/0ebcfd29e31f86345243c124, 2010.

[7]吴豪. zju1610 Count the Colors[OL]. https://www.360docs.net/doc/ce7628477.html,/21469854_d.html, 2008.

6广度优先搜索，俗称暴力搜索。

- 12 -

The Application of Segment Tree on Algorithm

Zhu Kaidi

zukaidi@https://www.360docs.net/doc/ce7628477.html,

Abstract

More and more problems about Statistics, Search, Planning, Sort, dyeing are appearing in the Computer Olympiad. Balanced Binary Tree and Segment Tree are two most common Data Structure to solve these problems. But do you feel tired of the High-Complexity Data Structure –Balanced Binary Tree? We can see that on some problems, the Segment Tree is a powerful alternative to the Balanced Binary Tree. This paper will mainly introduce the operation, optimization and application of the Segment Tree. This paper will also give the general solution to the dyeing problem.

Keywords: Segment Tree, Data Struct, Computer Science, Algorithm.

作者简介

本文作者朱凯迪是宁波工程学院计科091班的学生。从高中开始参加信息学奥林匹克竞赛，对算法非常感兴趣，同时也在高中打下了一定的算法基础。在大学期间继续参加大学生ACM程序设计大赛，更加深了对算法的了解。本文是建立在作者现有的算法基础上写的。

- 13 -