[算法总结] LCA倍增法 dfs

步骤

• 预处理 出 每个的2^j 的祖先
• 将两个点 移到同一位置
• 倍增从大到小的跳跃

模板

#include <iostream>
#include <vector>
#include <map>
#include <cstring>
#include <queue>
#include <set>
#include <algorithm>
using namespace std;
#define IOS  ios::sync_with_stdio(false);
#define CIT  cin.tie(0);
#define COT  cout.tie(0);

#define ll long long
#define x first
#define y second
#define pb push_back
#define endl '\n'
#define all(x) (x).begin(),x.end()
#define Fup(i,a,b) for(int i=a;i<=b;i++)
#define Fde(i,a,b) for(int i=a;i>=b;i--)

typedef priority_queue<int,vector<int>,greater<int>>  Pri_m;
typedef pair<int,int> pii;
typedef vector<int> VI;
map<int,int> mp;
const int N = 5e5+10 ;
int fa[N][31];//表示x这个点走到了2^k的点
int n,m;
int dep[N];
vector<int> g[N];
// int cost[N][31];  代价数组 可以用于最短路径查询
void dfs(int u,int father){
	fa[u][0] = father;// u往上走1位,那么就是他的父亲节点
	dep[u] = dep[fa[u][0]] + 1; //深度为父节点+1

	//第2^i 的祖先节点 是 第2^(i-1)的祖先节点的 2^(i-1)的祖先节点
	//节点从 小到大 1 2 3 4 5 也就是从树根倒叶子
	for(int i = 1; i<31;i++){
		fa[u][i] = fa[fa[u][i-1]][i-1];
// 		cost[u][i] = cost[fa[u][i-1]][i-1]+cost[u][i-1]
	}


	for(auto x : g[u]){
		if(x == father) continue;
// 		cost[x][0]  = w[u][x]
		dfs(x,u);
	}
}
int lca(int x,int y){
	if(dep[x] > dep[y]) swap(x,y);
	//令y的深度大于x

	int temp = dep[y] - dep[x];
	int ans =0  ;

	//将这个差进行二进制拆分
	for(int j = 0 ; temp ; ++j , temp >>=1){
		if(temp&1){
// 		ans+=cost[y][i]
			y =  fa[y][j];
		}
	}

	//如果 他们就是他们自己的祖先
	if(y == x) return y;

	for(int j= 30; j>= 0  && y!=x ;  -- j){
		if(fa[x][j]!=fa[y][j]){
// 			ans+= cost[x][y] + cost[y][j]
			x = fa[x][j];
			y = fa[y][j];
		}
	}
// 	ans+=cost[x][0]+cost[y][0]
	return fa[y][0];
}

void solve(){
	cin>>n;
	int s = 0 ;

	for(int i=1;i<=n;i++){
		int a,b;cin>>a>>b;
		if(b== -1){
			s = a;
			continue;
		}
		g[a].pb(b);
		g[b].pb(a);
	}
	dfs(s,0);
	cin>>m;
	for(int i=1;i<=m;i++){
		int a,b;cin>>a>>b;
		int lcaa = lca(a,b);
		if(lcaa == a)cout<<1<<endl;
		else if(lcaa == b) cout<<2<<endl;
		else cout<<0<<endl;

	}
}

int main(){
    //int t;cin>>t;while(t--)
    solve();
    return 0 ;
}

题目

https://www.luogu.com.cn/problem/P3379

#include <bits/stdc++.h>
using namespace std;
const int N = 5e6+10;
int h[N],e[N],ne[N],idx;
int f[N][21],dep[N];
int n,m,s;
void add(int a,int b )
{
    e[idx] = b,ne[idx] = h[a],h[a] = idx++;
}

void dfs(int u,int fa)///O nlogn
{
    dep[u] = dep[fa]+1;
    f[u][0]  =fa;

    for(int i = 0 ;i<=19;i++)
    f[u][i+1] = f[f[u][i]][i];

    ///f[i][j] i节点的第2^j代祖先

    for(int i =  h[u]; i!=-1; i=ne[i])
    {
        int  j  = e[i];
        if(j!=fa)
            dfs(j,u);
    }
}

int lca(int u,int v)
{
    if(dep[u]<dep[v]) swap(u,v);
    int d = dep[u] - dep[v];

    for(int  i = 0 ; (1<<i)<=d; i++)
        if((1<<i)&d)
            u=f[u][i];

    if(u!=v)
    {
        //  for(int  i = (int)log(n); i>=0; i -- )
        for(int i = 20; i>=0; i--)
            if(f[u][i] != f[v][i])
            {
                u  =f[u][i];
                v  =f[v][i];
            }
        return f[u][0];
    }
    //cout<<"none"<<endl;
    return u;
}

void solve()
{
    cin>>n>>m>>s;

    memset(h,-1,sizeof h );
    n--;
    while(n -- )
    {
        int a,b;
        cin>>a>>b;
        add(a,b),add(b,a);
    }

    dfs(s,0);
   // cal();

    while( m -- )
    {
        int a,b;
        cin>>a>>b;
        cout<<lca(a,b)<<endl;
    }
}
int main()
{
    ios::sync_with_stdio(false);
    solve();
    return 0;
}