LCA

引入

在一棵树中，要求X->Y的距离，怎么求呢？可以用SPFA。
那把问题升级一下，有Q次询问呢？SPFA或floyd就肯定会超时了。
注意是在一棵树中，所以我们有了一种新算法：LCA（最近公共祖先）。
要求最近公共祖先，从两个点一起往上跳，如果跳在了同一个点，那么这个点就是他们的LCA。
如图：

但是一步一步跳太慢了，我们可以用倍增的思想，∵任何数都可以拆为若干个2的幂的形式，∴我们可以一次跳2的X次幂步。我们按从大到小的幂来跳，但这样会出现一个问题，就是一次性跳到了根节点，∴我们跳到他们不相交为止，那么他们的LCA就是他们的父亲。
如图：

上面说的都是他们深度相同时，如果深度不同，就先把深度大的点先跳到同一深度即可。因此我们需要先用因此DFS来初始化深度和往上跳2的X次幂跳到的位置。
那么LCA跟X->Y的距离有什么关系呢？
如图：

根据容斥原理就可以知道，X到Y的距离就是：X的深度+Y的深度-2*LCA的深度。

例题

2528: 点的距离
题目描述
给定一棵树有n个节点的树，q个询问，每次询问点x到点y之间的距离
输入
第一行一个n，表示这棵树有n个节点。接下来n-1行，每行两个整数x,y表示x，y之间有一条连边。然后一个整数q，表示有q个询问，接下来q行，每行两个整数x，y
输出
输出q行，每行为一个询问结果
样例输入
6
1 2
1 3
2 4
2 5
3 6
2
2 6
5 6
样例输出
3
4
提示
n,q<=100000

分析

LCA裸题。

代码

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define full(a,b) memset(a,b,sizeof a)
#define N 100000+5
#define E 2*N
#define LOG 25
int n,q;
int ecnt,head[N],to[E],nxt[E];
int f[N][LOG+5],dep[N];
void add(int u,int v)
{
	to[++ecnt]=v;
	nxt[ecnt]=head[u];
	head[u]=ecnt;
}
void init()
{
	full(head,-1);
}
void pre(int u,int fa)//初始化 
{
	dep[u]=dep[fa]+1;//记录深度 
	f[u][0]=fa;//记录父亲 
	for(int i=0; i<=LOG; i++)
		f[u][i+1]=f[f[u][i]][i];//倍增跳 
	for(int i=head[u]; i!=-1; i=nxt[i])
	{
		int v=to[i];
		if(v==fa) continue;
		pre(v,u);
	}
}
int LCA(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);//假设X深度大 
	for(int i=LOG; i>=0; i--)
	{
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];//跳到同一个点 
		if(x==y) return x;//位置相同，X就是LCA 
	}
	for(int i=LOG; i>=0; i--)
		if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];//倍增跳 
	return f[x][0];//父亲是LCA 
}
int main()
{
//	freopen("2528: 点的距离.in","r",stdin);
//	freopen("2528: 点的距离.out","w",stdout);
	init();
	scanf("%d",&n);
	for(int i=1; i<n; i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		add(a,b),add(b,a);
	}
	pre(1,1);
	scanf("%d",&q);
	for(int i=1; i<=q; i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		printf("%d\n",dep[x]+dep[y]-2*dep[LCA(x,y)]);//公式 
	}
	return 0;
}

带边权LCA

例题

P2912 [USACO08OCT]Pasture Walking G

分析

再开一个dist数组记录X到根节点的边权和。注意！不能用dep代替dist数组。

代码

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define full(a,b) memset(a,b,sizeof a)
#define N 1000+5
#define E 2*N
int n,q;
int head[N],ecnt,to[E],nxt[E],cost[E];
int dep[N],f[N][25],dist[N];
void init()
{
    full(head,-1);
}
void add(int u,int v,int w)
{
    to[++ecnt]=v;
    nxt[ecnt]=head[u];
    head[u]=ecnt;
    cost[ecnt]=w;
}
void pre(int u,int fa)
{
    dep[u]=dep[fa]+1;
    f[u][0]=fa;
    for(int i=0; i<=20; i++)
        f[u][i+1]=f[f[u][i]][i];
    for(int i=head[u]; i!=-1; i=nxt[i])
    {
        int v=to[i];
        if(v==fa) continue;
        dist[v]=dist[u]+cost[i];
        pre(v,u);
    }
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=20; i>=0; i--)
    {
        if(dep[f[x][i]]>=dep[y]) x=f[x][i];
        if(x==y) return x;
    }
    for(int i=20; i>=0; i--)
        if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
int main()
{
//  freopen("[USACO2008OCT」Pasture Walking.in","r",stdin);
//  freopen("[USACO2008OCT」Pasture Walking.out","w",stdout);
    init();
    scanf("%d%d",&n,&q);
    for(int i=1; i<n; i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    pre(1,0);
    for(int i=1; i<=q; i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d\n",dist[a]+dist[b]-2*dist[LCA(a,b)]);
    }
    return 0;
}