码迷,mamicode.com
首页 > 编程语言 > 详细

算法导论学习-线段树(2)

时间:2015-02-06 13:03:14      阅读:275      评论:0      收藏:0      [点我收藏+]

标签:

线段树(1)http://www.cnblogs.com/fu11211129/p/4230000.html

1. 线段树应用之动态点插与统计:

--------------------------------

线段树(1)中讲的应用是区段的插值与统计,我们在线段树结构体中接入cover之一域,cover等于0表示该节点所代表的区域并没有被完全覆盖,cover大于等于1表示该节点所代表区域已经被完全覆盖,用线段树(1)博客里面的图来说明一下:

         技术分享         技术分享

如上图所示,我们最后统计的时候是找出cover不为零的节点,(再上图对应的是节点[3,4]. [5,6], [2], [4], [5], [7].),但是注意到[3,4]和[4] 还有[5,6]和[5]不能从重复统计,所以最后统计的长度是[2]+[3,4]+[5,6]+[7]=6.所以四根木条在墙上的总投影长是6. 

现在我们可以来归纳一下区段插值的几个重要部分:首先是插入动作,比如插入[L, R]线段,插入动作需要递归的从根节点开始插入,而递归的出口就是找到能够和[L, R]完全匹配的线段树节点,即找到a[i].left==L AND a[i].right==R,并且同时将a[i].cover值加1.,递归终止。然后是删除动作,假设删除[L, R],同样的也是要从根节点开始删除,也是要递归的找到与[L, R]完全匹配的线段树节点a[i],并且将a[i].cover值减去1.最后是统计,仍然是从根节点开始递归的统计,递归的出口要么是找到的节点a[i]满足a[i].cover>=0,则返回该节点代表的区段长度。要么是找到叶子节点(a[i].left==a[i].right),如果叶子节点的cover不为0,则返回1,否则返回零。

以上是对区段插入的回顾,下面开始介绍动态点插与统计。所谓动态点插(其实是我自己DIY的,所以我就用图表示一下吧,大家理解就好)

技术分享

如上图所示,1-8表示相互连接的8个槽,然后在这些草内的任意位置放置任意数量个小球,然后让你求出任意区段内[L, R]的小球数量。比如[2,3]区段内的小球数量是2,[1,6]区段内的小球数量是6。要解决这个问题,可以优良中办法,一种方法不用说也知道,暴力嘛,开一个数组A,然后更新数组的值。。这种做法当“槽”的范围过大时是效率很低的。所以这时候线段树再次派上用场了。我们还是用cover域来表示某个节点缩地阿彪范围的小球数量。比如根节点代表1-8槽的小球数量。

###第一步,建立线段树(http://www.cnblogs.com/fu11211129/p/4230000.html 里面有介绍,这里直接跳过)

###第二步,线段树插入(点插)

1 void updatePoint(int pos,int step){
2     a[step].num++;
3     if(a[step].left==a[step].right&&a[step].left==pos) return;//leaf node
4     int mid=(a[step].left+a[step].right)/2;
5     if(pos<=mid) updatePoint(pos,step*2);
6     else updatePoint(pos,step*2+1);
7 }

我们可以看到,线段树点插动作是“一路插值的”,这也不难理解,比如我在第3个槽内放置一个球,那么显然的,节点[3].num=1,但同时的,节点[3,4],[1,2,3,4],[1,2,3,4,5,6,7,8]的num值也是为1.所以线段树点插可以理解记忆为:一路递归向下插值。

###第三步,线段树统计

 1 int query(int L,int R,int step){
 2     if(L==a[step].left&&R==a[step].right){
 3         return a[step].num;
 4     }
 5     else{
 6         int mid=(a[step].left+a[step].right)/2,ans=0;
 7         if(R<=mid) ans+=query(L,R,step*2);
 8         else if(L>=mid+1) ans+=query(L,R,step*2+1);
 9         else{
10             ans+=query(L,mid,step*2)+query(mid+1,R,step*2+1);
11         }
12         return ans;
13     }
14 }

可以看到,线段树点插结果统计是按照区段来查询的,所以递归查询的出口也就是找到与之匹配的节点,然后返回相应的cover值,对应代码2-4行。如果与当前节点不匹配,则递归的“往下”查[6-8],如果带查询区段在左右子树都有子结果,则返回两个子结果之和。

2. 线段树应用之区间最大最小值问题:

-----------------------------------

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 using namespace std;
 5 const int max_size=100;
 6 const int inf=1<<30;
 7 struct segmentTree{
 8     int lc,rc,mi,mx;
 9 }a[max_size*3];
10 void buildTree(int L,int R,int rt){
11     a[rt].lc=L;
12     a[rt].rc=R;
13     a[rt].mi=inf;
14     a[rt].mx=-inf;
15     if(L==R) return;
16     int mid=(a[rt].lc+a[rt].rc)/2;
17     buildTree(L,mid,rt*2);
18     buildTree(mid+1,R,rt*2+1);
19 }
20 void insert(int pos,int v,int rt){
21     if(pos==a[rt].lc&&a[rt].lc==a[rt].rc){
22         a[rt].mi=v;
23         a[rt].mx=v;
24         return;
25     } 
26     int mid=(a[rt].lc+a[rt].rc)/2;
27     if(pos<=mid) insert(pos,v,rt*2);
28     else insert(pos,v,rt*2+1);
29     a[rt].mi=min(a[rt*2].mi,a[rt*2+1].mi);
30     a[rt].mx=max(a[rt*2].mx,a[rt*2+1].mx);
31 }
32 int query_min(int L,int R,int rt){
33     if(L==a[rt].lc&&R==a[rt].rc) return a[rt].mi;
34     if(a[rt].lc<a[rt].rc){
35         int mid=(a[rt].lc+a[rt].rc)/2,ret;
36         if(R<=mid) ret=query_min(L,R,rt*2);
37         else if(L>=mid+1) ret=query_min(L,R,rt*2+1);
38         else ret=min(query_min(L,mid,rt*2),query_min(mid+1,R,rt*2+1));
39         return ret;
40     }
41 }
42 int main(){
43     int n,m,q,p,v;
44     while(scanf("%d",&n)!=EOF){
45         buildTree(1,n,1);
46         scanf("%d",&m);
47         for(int i=1;i<=m;i++){
48             scanf("%d%d",&p,&v);
49             insert(p,v,1);
50         }
51         scanf("%d",&q);
52         while(q--){
53             int a,b;
54             scanf("%d%d",&a,&b);
55             printf("%d\n",query_min(a,b,1));
56         }
57     }
58 }

 

算法导论学习-线段树(2)

标签:

原文地址:http://www.cnblogs.com/fu11211129/p/4231803.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!