【模板】最近公共祖先(LCA)

树的简介

树常见用来表示家族谱的家族关系,他像一个倒着的树。他的构成为:

  • 根节点( r o o t root root ): 所有节点都由一个 根节点( r o o t root root ,且只有一个 根节点( r o o t root root
  • 父亲节点( f a t h e r − n o d e father-node fathernode ): 每个 父亲节点( f a t h e r − n o d e father-node fathernode 都有一个或者多个 孩子节点( c h i l d − n o d e child-node childnode
  • 孩子节点( c h i l d − n o d e child-node childnode ): 每个 孩子节点( c h i l d − n o d e child-node childnode 都有一个 父亲节点( f a t h e r − n o d e father-node fathernode 或者 根节点( r o o t root root

二叉树 - 特例

二叉树和树基本一样,但是他也和树有一点区别:

  • 每一个 父亲节点( f a t h e r − n o d e father-node fathernode 最多只能有两个或者没有 孩子节点( c h i l d − n o d e child-node childnode
  • 两个 孩子节点( c h i l d − n o d e child-node childnode 分别为 左孩子节点( l e f t − c h i l d − n o d e left-child-node leftchildnode右孩子节点( r i g h t − c h i l d − n o d e right-child-node rightchildnode ,但是一个二叉树最多只能有两个孩子节点,也可以没有孩子节点

树的存储

树的存储经常使用 邻接表 进行存储,他的存储如下表格:

如果 1 1 1 2 2 2 的父亲节点,那么 2 2 2 就是 1 1 1 的孩子节点:

1 2 3
1 f a l s e false false t r u e true true f a l s e false false
2 t r u e true true f a l s e false false f a l s e false false
3 f a l s e false false f a l s e false false f a l s e false false

我们在代码里可以用 v e c t o r vector vector 来存储:

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5+10;
vector<int> G[N];
struct node {
    int v,id;
}
vector<node> Q[N];

int main() {
    int n,m;
    scanf("%d%d",&n,&m);
    int u,v;
    for(int i=1;i<n;i++) {
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    for(int i=1;i<=m;i++) {
        scanf("%d%d",&u,&v);
        Q[u].push_back(node{v,i});
        Q[v].push_back(node{u,i});
    }
    return 0;
}

这样可以用 v e c t o r vector vector 进行存储,我们也可以用 数组 进行存储:

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5+10;
bool a[N][N];
int b[N][N];

int main() {
    int n,m;
    scanf("%d%d",&n,&m);
    int u,v;
    for(int i=1;i<n;i++) {
        scanf("%d%d",&u,&v);
        a[u][v] = true;
        a[v][u] = true;
    }
    for(int i=1;i<=m;i++) {
        scanf("%d%d",&u,&v);
        a[u][i] = v;
        a[v][i] = u;
    }
    return 0;
}

我们也有可以用 t a r j a n tarjan tarjan 算法进行查找:

int find(int x) {	//截枝查找(折半查找)
    if(fa[x] == x) {
        return x;
    }
    return fa[x] == find(fa[x]);
}

void tarjan(int u) {
    vis[u] = 1;
    for(auto v : G[u]) {
        if(!vis[v]) {
            tarjan(v);
            fa[v] = u;
        }
    }
    for(auto e : Q[u]) {
        int v = e.v;
        int id = e.id;
        if(vis[v]) {
            ans[id] = find(v);
        }
    }
}

所以我们可以写出 洛谷的【模板】最近公共祖先(LCA) 代码:

#include <bits/stdc++.h>
using namespace std;

const int N = 5e5+5;
vector<int> G[N];
struct node {
	int v,id;
};
vector<node> Q[N];
int fa[N];
bool vis[N];
int ans[N];

int find(int x) {
	if(x == fa[x]) {
		return x;
	}
	return fa[x] = find(fa[x]);
}

void tarjan(int u) {
	vis[u] = 1;
	for(auto v : G[u]) {
		if(!vis[v]) {
			tarjan(v);
			fa[v] = u;
		}
	}
	for(auto e : Q[u]) {
		int v = e.v;
		int id = e.id;
		if(vis[v]) {
			ans[id] = find(v);
		}
	}
}

int main()
{
	int n,m,s;
	scanf("%d%d%d",&n,&m,&s);
	int u,v;
    // 使用vector进行存储
	for(int i=1;i<n;i++) {
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	for(int i=1;i<=m;i++) {
		scanf("%d%d",&u,&v);
		Q[u].push_back(node{v,i});
		Q[v].push_back(node{u,i});
	}
	for(int i=1;i<=n;i++) {
		fa[i] = i;
	}
    // 使用 tarjan算法 进行查找
	tarjan(s);
    // 输出
	for(int i=1;i<=m;i++) {
		printf("%d\n",ans[i]);
	}
	return 0;
}

这样我们就可以完成洛谷的 【模板】最近公共祖先(LCA)

相关推荐

  1. 模板最近公共祖先LCA

    2024-04-29 19:10:06       34 阅读
  2. 最近公共祖先lca)倍增算法【模板

    2024-04-29 19:10:06       51 阅读
  3. 最近公共祖先LCA

    2024-04-29 19:10:06       42 阅读
  4. 最近公共祖先LCA)主要算法:

    2024-04-29 19:10:06       50 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-04-29 19:10:06       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-04-29 19:10:06       100 阅读
  3. 在Django里面运行非项目文件

    2024-04-29 19:10:06       82 阅读
  4. Python语言-面向对象

    2024-04-29 19:10:06       91 阅读

热门阅读

  1. 逻辑填空。

    2024-04-29 19:10:06       27 阅读
  2. 【星海出品】前端和后端交互的python EEL小技巧

    2024-04-29 19:10:06       27 阅读
  3. centos学习-掌握核心命令之-yum

    2024-04-29 19:10:06       29 阅读
  4. 【二叉树算法题记录】101. 对称二叉树

    2024-04-29 19:10:06       29 阅读
  5. 【Docker】常见命令汇总

    2024-04-29 19:10:06       35 阅读
  6. 力扣56. 合并区间

    2024-04-29 19:10:06       33 阅读