实用算法实现-第6篇线段树

来源：互联网发布：买一个域名要多少钱编辑：程序博客网时间：2024/05/01 11:23

6.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、以点为结点。2、以线段为结点。区别如图：上面一个以线段为结点，下面一个以点为结点：

对线段树存在：

定理：线段树把区间上的任意一条线段都分成不超过2logL条线段。

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

对线段树的可以进行扩展。

1. 测度。结点所表示区间中线段覆盖过的长度，存储在各结点中。

2. 独立线段数。区间中互不相交的线段条数。

3. 权和。区间所有元线段的权和。

测度的递推公式如下：

a[j] - a[i] 该结点 Count>0

M = 0 该结点为叶结点且 Count=0

Leftchild ↑ .M + Rightchild ↑ .M 该结点为内部结点且 Count=0连续段数

这里的连续段数指的是区间的并可以分解为多少个独立的区间。如 [1 ， 2] ∪[2，3]∪ [5 ， 6] 可以分解为两个区间[1 ， 3] 与 [5 ， 6] ，则连续段数为 2 。增加一个数据域 Lines_Tree.line 表示该结点的连续段数。 Line 的讨论比较复杂，内部结点不能简单地将左右孩子的 Line 相加。所以再增加 Lines_Tree.lbd 与 Lines_Tree.rbd 域。定义如下：

1 左端点 I 被描述区间盖到

lbd =

0 左端点 I 不被描述区间盖到

1 右端点 J 被描述区间盖到

rbd =

0 右端点 J 不被描述区间盖到

lbd 与 rbd 的实现：

1 该结点 count > 0

lbd = 0 该结点是叶结点且 count = 0

leftchild ↑ .lbd 该结点是内部结点且 Count=0

1 该结点 count > 0

rbd = 0 该结点是叶结点且 count = 0

rightchild ↑ .rbd 该结点是内部结点且 Count=0

有了 lbd 与 rbd ， Line 域就可以定义了：

1 该结点 count > 0

Line = 0 该结点是叶结点且 count =0

Leftchild ↑ .Line + Rightchild ↑.Line - 1 当该结点是内部结点且 Count=0 ， Leftchild ↑ .rbd = 1 且 Rightchild ↑ .lbd = 1

Leftchild ↑.Line + Rightchild ↑ .Line 当该结点是内部结点且 Count=0 ， Leftchild ↑ .rbd 与 Rightchild ↑ .lbd 不都为1

6.2 利用线段树实现区间的动态插入和删除

6.2.1 实例

PKU JudgeOnline, 1151, Atlantis.

6.2.2 问题描述

在二维平面分部着一些矩形，矩形有可能重合。求矩形的总面积。

6.2.3 分析

这个题在《算法艺术与信息学竞赛》中第一章介绍数据结构时，讲到线段树的时候有解题分析。

用线段树来记载纵向上是不是被覆盖，用测度来表示区间中被覆盖了多少长度。

为了降低复杂度，可以将坐标离散化，如下图所示：

从左到右扫描长方形的左侧边和右侧边，如果是左侧边则加入线段树中，否则从线段书中删除。同时用横向扫描的距离乘以线段树的测度，就得到了扫描过了的被覆盖的面积。

本题和PKU JudgeOnline,1117, Picture题都对线段树进行了扩展。本题只用到了测度的扩展，而1117题还用到了独立线段数的扩展。

6.2.4 程序

//离散化+ 线段树+ 扫描线//本题与JudgeOnline 1177 picture 极相似,现在回想起来甚至比1177 还要简单一些.与1177 不同的是本题中的坐标是浮点//类型的故不能将坐标直接离散.我们必须为它们建立一个对应关系,用一个整数去对应一个浮点数//这样的对应关系在本题的数组y[] 中#include<iostream>#include<algorithm>#include<cmath>#include<iomanip>using namespace std; struct node{     int st, ed,c;   //c : 区间被覆盖的层数,m: 区间的测度     double m;}ST[802];struct line{     doublex,y1,y2;   //纵方向直线, x:直线横坐标, y1 y2:直线上的下面与上面的两个纵坐标     bools;     //s = 1: 直线为矩形的左边, s = 0:直线为矩形的右边}Line[205];double y[205],ty[205]; //y[] 整数与浮点数的对应数组;ty[]:用来求y[]的辅助数组 void build(int root, int st, int ed){     ST[root].st = st;     ST[root].ed = ed;     ST[root].c = 0;     ST[root].m = 0;     if(ed - st> 1){         int mid= (st+ed)/2;         build(root*2, st, mid);         build(root*2+1, mid, ed);     }}inline void updata(int root){     if(ST[root].c> 0)         //将线段树上区间的端点分别映射到y[]数组所对应的浮点数上,由此计算出测度         ST[root].m = y[ST[root].ed-1] -y[ST[root].st-1];     else if(ST[root].ed - ST[root].st == 1)         ST[root].m = 0;     elseST[root].m = ST[root*2].m + ST[root*2+1].m;}void insert(int root, int st, int ed){     if(st <=ST[root].st && ST[root].ed <= ed){         ST[root].c++;         updata(root);         return;     }     if(ST[root].ed- ST[root].st == 1)return ;//不出错的话这句话就是冗余的     int mid =(ST[root].ed + ST[root].st)/2;     if(st <mid)         insert(root*2, st, ed);     if(ed >mid)         insert(root*2+1, st, ed);     updata(root);}void Delete(int root, int st, int ed){     if(st <=ST[root].st && ST[root].ed <= ed){         ST[root].c--;          updata(root);         return;     }     if(ST[root].ed- ST[root].st == 1)return ; //不出错的话这句话就是冗余的     int mid =(ST[root].st + ST[root].ed)/2;     if(st <mid)         Delete(root*2, st, ed);     if(ed >mid)         Delete(root*2+1, st, ed);     updata(root);}int Correspond(int n, double t){     //二分查找出浮点数t 在数组y[]中的位置(此即所谓的映射关系)     intlow,high,mid;     low = 0; high = n-1;     while(low< high){         mid = (low+high)/2;         if(t> y[mid])              low = mid + 1;         elsehigh = mid;     }     returnhigh+1;}bool cmp(line l1, line l2){     return l1.x< l2.x;} int main(){     intn,i,num,l,r,c=0;     doublearea,x1,x2,y1,y2;     while(cin>>n,n){         for(i =0; i < n; i++){              cin>>x1>>y1>>x2>>y2;              Line[2*i].x = x1; Line[2*i].y1 =y1; Line[2*i].y2 = y2; Line[2*i].s = 1;              Line[2*i+1].x = x2; Line[2*i+1].y1= y1; Line[2*i+1].y2 = y2; Line[2*i+1].s = 0;              ty[2*i] = y1; ty[2*i+1] = y2;         }         n <<= 1;         sort(Line, Line+n, cmp);         sort(ty, ty+n);         y[0] = ty[0];         //处理数组ty[]使之不含重覆元素,得到新的数组存放到数组y[]中         for(i=num=1;i < n; i++)              if(ty[i]!= ty[i-1])                   y[num++] = ty[i];         build(1, 1, num); //树的叶子结点与数组y[]中的元素个数相同,以便建立一一对应的关系         area = 0;         for(i =0; i < n-1; i++){              //由对应关系计算出线段两端在树中的位置              l = Correspond(num, Line[i].y1);              r = Correspond(num, Line[i].y2);              if(Line[i].s)//插入矩形的左边                   insert(1, l, r);              else    //删除矩形的右边                   Delete(1, l, r);              area += ST[1].m * (Line[i+1].x -Line[i].x);         }         cout<<"Testcase #"<<++c<<endl<<"Totalexplored area: ";         cout<<fixed<<setprecision(2)<<area<<endl<<endl;     }     return 0;}

6.3 计算数组区间第K大的数

PKU JudgeOnline, 2761, Feed the dogs则是线段树的另外一个应用：实用线段树来计算数组区间[i, j]中元素第k小（或第K大）的数。只要添写一个函数，根据线段树中每个结点的覆盖树木来判断第k大的树是哪一个。

当初始化，或者区间[i, j]发生变化时，需要对线段树进行添加或者删除操作。每当增加（或删除）一个大小为X的点时，就在树上添加（或删除）一条(X,MaxLen)的线段（不含端点），当要查询一个点的排名时，只要看看其上有多少条线段就可以了。

int query(int root, intcount){     if(count<= ST[root].c){         returnST[root].st;     }else if(ST[root].ed - ST[root].st == 1){         returnST[root].ed;     }     count -= ST[root].c;     if(count<= ST[root*2+1].c){         returnquery(root*2, count);     }else{         returnquery(root*2+1, count);     }}

1.4 实例

PKU JudgeOnline, 1151, Atlantis.

PKU JudgeOnline, 1117, Picture.

PKU JudgeOnline, 2761, Feed the dogs.

PKU JudgeOnline, 2528, Mayor'sposters.

本文章欢迎转载，请保留原始博客链接http://blog.csdn.net/fsdev/article