0901-倍增法求LCA+例题

一直听大佬们说男生用树链剖分求lca，只有女生用倍增？？？？然而姑娘我表示今天之前一直不会倍增，为了显示我女生的身份我决定学习一下倍增，结果第一次写的时候漏洞百出啊，不过还好有可耐的勾勾同学耐心纠错，感谢Orz，所以在这儿打个总结

【前言】

LCA（Least Common Ancestors）的意思是最近公共祖先，即在一棵树中，找出两节点最近的公共祖先。

这个图中，6,8的lca就是1；12和8的lca就是8

倍增，字面意思就是“成倍增长”。倍增总和二进制划分密切相关，ST表和LCA都运用到了倍增的思想（ST表之前写过的这儿）

 

【正文】

依然是这个图，如果没学倍增的话，暴力做法就是先把 x ， y 调到同一高度，然后同时往上走，走到的第一个相同的点，就是他们的最近公共祖先了，这样的话，看起来就会执行很多次，我们在此基础上搞一下，每次都跳 2^i 步，就有了倍增这个玩意儿

我们先定义fa [ x ] [ i ] 表示 x 号节点向上走 2^i 步所到达的祖先，那么显然 fa [ x ] [ 0 ] 就是 x 的父亲，（往上走 2^0 = 1 步） 

而其他的我们则用这个递推公式进行计算： fa [ x ] [ i ] = fa [ fa [ x ] [ i - 1] [ i - 1 ] （不懂的同学可以看图哦）

预处理部分搞定，哦，还别忘了要在第一次dfs的时候顺便把每个点的dep（深度）求出来

现在来说怎么求lca

我们先把两个点中深度较大的那一个，往上移，直到和另一点深度一样

这个时候如果移动过后的两个点重合了，那说明这两个在同一条链上，则深度较浅的那一个就是他们两的lca

否则我们就开始“跳”了~

先从最大的可能跳着走，（也就是说跳到最远的地方，然后往回跳），如果跳到位置是一样的，我们就减小一半再跳，直到遇到两个点不一样，我们就把 x 和 y 移到那两个位置上去，然后接着跳，直到循环结束，最后 x 和 y 的lca就是退出循环时那个 x 的父亲，为什么最后可以跳到lca的位置呢？可以这样来想，每一个位置都可以由2的多少次方累加得到（可能口吃不太清楚，大家如果不明白可以等会儿对着代码模拟一遍，就会恍然大悟其正确性了）

 

例题

来我们看一道题，感受一下倍增的魅力

传送门

 

大致题意

给你一颗树，求任意两点之间的距离

分析

我们用d[i] 表示根节点到 i 号节点的距离（由于这道题没有说明要求，我们就默认1号节点为根吧）

那么如果题目给出来的两个点：x，y 在同一条链上，我们交换x，y 使得 x 的深度大一些 则ans = d [ x ] - d [ y ]

若不再一条链上，我们倍增找到x和y的lca，然后ans = d [ x ] +d [ y ] - 2*d[ lca ] ,注意是减去2倍的，因为到x，到y都经过了一遍

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#define N 100009///
#define in read()
using namespace std;
int T,n,m,head[N],nxt[N],to[N],w[N],tot;
int fa[N][23],depth[N],d[N];
void add(int x,int y,int z){nxt[++tot]=head[x];head[x]=tot;to[tot]=y;w[tot]=z;}
inline int read(){
	char ch;int f=1,res=0;
	while((ch=getchar())<'0'||ch>'9')	if(ch=='-') f=-1;
	while(ch>='0'&&ch<='9'){
		res=(res<<3)+(res<<1)+ch-'0';
		ch=getchar();
	}
	return f==1?res:-res;
}
void getfn(int u,int fu){
	fa[u][0]=fu;
	for(int i=head[u];i;i=nxt[i]){
		int v=to[i];
		if(depth[v]||fu==v) continue;
		depth[v]=depth[u]+1;d[v]=d[u]+w[i];
		getfn(v,u);
	}
}
int lca(int x,int y){
	if(depth[x]<depth[y]) swap(x,y);
	int dper=d[x],sha=d[y];
	for(int i=20;i>=0;--i)
		if(depth[fa[x][i]]>=depth[y]) x=fa[x][i];
	if(x==y) return dper-d[y];
	for(int i=20;i>=0;--i)
	{
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return dper+sha-2*d[fa[x][0]];///2倍哦
}
int main(){
	T=in;
	while(T--){
		n=in;m=in;
		int i,j,k;
		memset(head,0,sizeof(head));
		memset(depth,0,sizeof(depth));
		memset(d,0,sizeof(d));memset(fa,0,sizeof(fa));
		tot=0;
		for(i=1;i<n;++i){
			int x,y,z;
			x=in;y=in;z=in;
			add(x,y,z);add(y,x,z);
		}
		depth[1]=1;
		getfn(1,0);
		for(j=1;j<=20;++j)由于我们是由j - 1推到 j ，那么肯定是递增求解
			for(i=1;i<=n;++i)
				fa[i][j]=fa[fa[i][j-1]][j-1];
		for(i=1;i<=m;++i){
			int x,y;
			x=in;y=in;
			printf("%d\n",lca(x,y));
		}
		//printf("\n");   ///哪位高人可以帮我翻译一下output的那一句话吗？？为什么不需要再输出一行空行呢？？
	}
	return 0;
}