学习线段树

参考文章：

http://blog.csdn.net/chinaeli/archive/2008/11/01/3201574.aspx

http://hi.baidu.com/czyuan_acm/blog/item/36f708cc23c1ec1501e9282c.html

http://www.cnblogs.com/sherlockhua/archive/2007/07/30/836789.html

 

从简单说起，线段树其实可以理解成一种特殊的二叉树。但是这种二叉树较为平衡，和静态二叉树一样，都是提前已经建立好的树形结构。针对性强，所以效率要高。这里又想到了一句题外话：动态和静态的差别。动态结构较为灵活，但是速度较慢；静态结构没有节省内存，但速度较快。算法导论中有涉及区间树的内容，那里用红黑树来进行维护。本文主要讲静态的线段树。

接着回到线段树上来，线段树是建立在线段的基础上，每个结点都代表了一条线段[a , b]。长度为1的线段成为元线段。非元线段都有两个子结点，左结点代表的线段为[a , (a + b ) / 2]，右结点代表的线段为[( a + b ) / 2 , b]。

长度范围为[1 , L] 的一棵线段树的深度为log ( L - 1 ) + 1。这个显然，而且存储一棵线段树的空间复杂度为O（L）。

线段树支持最基本的操作为插入和删除一条线段。下面已插入为例，详细叙述，删除类似。

将一条线段[a , b] 插入到代表线段[l , r]的结点p中，如果p不是元线段，那么令mid＝（l＋r）/2。如果a<mid，那么将线段[a , b] 也插入到p的左儿子结点中，如果b>mid，那么将线段[a , b] 也插入到p的右儿子结点中。

插入（删除）操作的时间复杂度为O （ Log n ）。

线段树其最简单的应用就是记录线段有否被覆盖，并随时查询当前被覆盖线段的总长度。那么此时可以在结点结构中加入一个变量int coverLength；代表当前结点代表的子树中被覆盖的线段长度和。这样就要在插入（删除）当中维护这个coverLength值，于是当前的覆盖总值就是根节点的coverLength值了。

另外也可以将coverLength换成bool cover；支持查找一个结点或线段是否被覆盖。

根据上面的讨论，我们设计线段树的节点结构为

class Node { public: Node(int lt, int rt) { if(rt < lt) { left = rt; right = lt; }else{ left = lt; right = rt; } count = covLen = lbd = rbd = numOfLines = 0; pleft = pright = NULL; } private: friend class LineTree; int left, right; int count; //线段覆盖条数 int covLen; //测度，该区间中被覆盖过线段的总长度。 int numOfLines; //该区间上连续线段的条数 int lbd, rbd; //用来计算连续段数,lbd: left bound, 0代表左边界没有被覆盖，相反为1. rbd同理。 Node * pleft; Node * pright; };

count:线段覆盖条数，当插入线段完全覆盖线段树上的区间时（ left<=NODE.left && NODE.right<=right ），count才增长，否则，就将该线段插入到子节点中。

covLen:测度，通俗点说，就是该区间中被覆盖过线段的总长度。当该区间的count>1时，covLen等于该区间的长度，如果是离散化之后的线段树，就加上对应的线段的长度。如果这个节点不是内部节点而是叶节点并且count=0，那covLen就等于0。如果这个节点的count等于0，但是他是内部节点，那么covLen就等于他左右子节点的covLen值的和。

numOfLines连续段数：就是该区间上连续线段的条数，比如线段树[1,10]上，有两条线段[1,5],[8,10]，那么[1,10]这个区间上的连续线段的条数就是2；如果有两条线段是[1,5],[5,10]，那么该区间上连续线段条数就是1。由此可以看出，连续线段条数，与该节点的左右子节点上的左右端点的覆盖情况有关，因此还有设立2个域lbd, rbd表示该区间的左右端点的覆盖情况。

• 假如count>0，说明这个区间被一条线段覆盖了，那么显然这个区间的numOfLines等于1；假如这个节点是叶节点并且count=0，那么numOfLines显然等于0；
• 假如这个区间的count=0，但是它是内部节点，就又要分2种情况：①左子节点的rbd和右子节点lbd都等于1, 即上例中[1,5],[5,10]的情况，那么numOfLines就等于1。 ②左子节点的rbd和右子节点lbd至少有1个不等于1, 即上例中[1,5],[8,10]的情况，那么numOfLines就等于2。对于lbd,rbd的定义，可以这样该区间的lbd等于他的左子节点的lbd，rbd等于他的右子节点的rbd。

---------------------------完整线段树代码------------------------------------------
class Node { public: Node(int lt, int rt) { if(rt < lt) { left = rt; right = lt; }else{ left = lt; right = rt; } count = covLen = lbd = rbd = numOfLines = 0; pleft = pright = NULL; } private: friend class LineTree; int left, right; int count; //线段覆盖条数 int covLen; //测度，该区间中被覆盖过线段的总长度。 int numOfLines; //该区间上连续线段的条数 int lbd, rbd; //用来计算连续段数,lbd: left bound, 0代表左边界没有被覆盖，相反为1. rbd同理。 Node * pleft; Node * pright; }; class LineTree { public: LineTree() { root = NULL;} void build(int lt, int rt); void insert(int lt, int rt); void del(int lt, int rt); void clear(); int getCoverLen() { return root->covLen; } int getNumOfLines() { return root->numOfLines; } int getCov() { return root->count; } ~LineTree() {clear();} private: Node * root; void init(Node * pnode = NULL) { root = pnode; } void updateCoverLen() { if(root->count > 0) { root->covLen = root->right - root->left; }else if( root->left + 1 == root->right) { root->covLen = 0; }else{ root->covLen = (root->pleft)->covLen + (root->pright)->covLen; } } void updateNumOfLines() { if(root->count > 0) { root->lbd = root->rbd = root->numOfLines = 1; }else if(root->left + 1 == root->right) { root->lbd = root->rbd = root->numOfLines = 0; }else{ root->lbd = (root->pleft)->lbd; root->rbd = (root->pright)->rbd; root->numOfLines = (root->pleft)->numOfLines + (root->pright)->numOfLines - (root->pleft)->rbd * (root->pright)->lbd; } } }; void LineTree::build(int lt, int rt) { root = new Node(lt, rt); if(rt - lt > 1) { int mid = lt + (rt - lt)/2; LineTree * tmp = new LineTree(); tmp->build(lt, mid); root->pleft = tmp->root; tmp->init(); tmp->build(mid, rt); root->pright = tmp->root; } } void LineTree::insert(int lt, int rt) { if(lt <= root->left && rt >= root->right) (root->count)++; else{ LineTree * tmp = new LineTree(); int mid = root->left + (root->right - root->left)/2; if(lt < mid) { tmp->init(root->pleft); tmp->insert(lt, rt); } if(rt > mid) { tmp->init(root->pright); tmp->insert(lt, rt); } } updateCoverLen(); updateNumOfLines(); } void LineTree::del(int lt, int rt) { if(lt <= root->left && rt >= root->right) (root->count)--; else{ LineTree * tmp = new LineTree(); int mid = root->left + (root->right - root->left)/2; if(lt < mid) { tmp->init(root->pleft); tmp->del(lt, rt); } if(rt > mid) { tmp->init(root->pright); tmp->del(lt, rt); } } updateCoverLen(); updateNumOfLines(); } void LineTree::clear() { if(root == NULL) return ; LineTree * tmp = new LineTree(); tmp->init(root->pleft); tmp->clear(); tmp->init(root->pright); tmp->clear(); delete root; root = NULL; } int main() { LineTree * t = new LineTree(); t->build(1, 8); t->insert(2,3); t->insert(4, 5); t->insert(6, 8); t->insert(1, 7); //t->clear(); return 0; }

理解代码后，可以到POJ 2777 Count Color 和 POJ 3264 Balanced Lineup去小试身手。