最近公共祖先——朴素LCA算法

在一些题目中经常要求求出树上两点的最短距离,想必写暴力搜索一定是会超时不少的,而且也是非常痛苦的(本人就作死过写暴力……)就像这种题目……

2370 小机房的树
 时间限制: 1 s
 空间限制: 256000 KB
 题目等级 : 钻石 Diamond
 题解
题目描述 Description
小机房有棵焕狗种的树,树上有N个节点,节点标号为0到N-1,有两只虫子名叫飘狗和大吉狗,分居在两个不同的节点上。有一天,他们想爬到一个节点上去搞基,但是作为两只虫子,他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量(从父亲节点爬到此节点也相同),他们想找出一条花费精力最短的路,以使得搞基的时候精力旺盛,他们找到你要你设计一个程序来找到这条路,要求你告诉他们最少需要花费多少精力

输入描述 Input Description
第一行一个n,接下来n-1行每一行有三个整数u,v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ,v 表示两只虫子所在的节点
输出描述 Output Description
一共有m行,每一行一个整数,表示对于该次询问所得出的最短距离。

样例输入 Sample Input
3

1 0 1

2 0 1

3

1 0

2 0

1 2

样例输出 Sample Output
1

1

2

 

数据范围及提示 Data Size & Hint
1<=n<=50000, 1<=m<=75000, 0<=c<=1000

看到底下的数据,顿时感觉自己就炸了……,所以我们不能什么都用暴力(当然比赛的时候该用还是得用),我们迫切需要一个新的算法,来求树上两点的最短距离。

最近公共祖先——LCA算法就可以做到这一点,由于本人水准比较低,所以只会朴素的写法……,请大家不要介意。

LCA,实际上是找出两个点的公共祖先,而后根据两个点的深度进行操作。所以首先我们要将读入的点建树记录父节点并且进行记录深度操作。

就上题而言,我们可以进行如下的操作进行记录深度……

 1 memset(dist,0,sizeof(dist));
 2 memset(d,0,sizeof(d));
 3 d[1] = 1;
 4 memset(vi,false,sizeof(vi));
 5 cin >> n;
 6 vi[1] = true;
 7 for(int i = 1;i < n;++i){
 8     cin >> u >> v >> c;
 9     u++;
10     v++;
11     if(!vi[v])swap(u,v);//因为我们并不知道到底应该用u还是v作为父节点,所以进行这一步操作,让已经被更新过的当父节点;
12     vi[u] = 1;
13     f[u] = v;
14     dist[u] = dist[v] + c;
15     d[u] = d[v] + 1;
16 }


记录完深度以后,我们就可以进行LCA的核心步骤——找最近公共祖先了,原理如下:

假设现在有树上A、B两个点,且两个点不一样。我们让深度大的那个向上走到他的父节点上去。等到他们两个深度相同而且还不是在一个点上的时候,我们让他们两个都往上跳,一直到两个点重合为止。核心代码如下:

 1 int lca(int a,int b){
 2     while(d[a] > d[b]){
 3         a = f[a];
 4     }
 5     while(d[a] < d[b]){
 6         b = f[b];
 7     }
 8     while(d[a] == d[b] && a != b){
 9         a = f[a];
10         b = f[b];
11     }
12     if(a == b){
13         return a;
14     }
15 }

这样子我们只是找到了他们的最近公共祖先是那个点,但我们并不知道距离。这样子就有两种算法:

一:两个点到最近公共祖先的距离之和

二:两个点到根的距离之和减去两倍的两点最近公共祖先到跟的距离。

这样朴素LCA就完结了。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值