LCA(最近公共祖先)模板
Tarjan版本
/* gyt
Live up to every day */
#pragma comment(linker,"/STACK:1024000000,1024000000")
#include<cstdio>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#include<stack>
#include<cstring>
#include<queue>
#include<set>
#include<string>
#include<map>
#include <time.h>
#define PI acos(-1)
using namespace std;
typedef long long ll;
typedef double db;
const int maxn = 1e5+;
const ll maxm = 1e7;
const ll base = ;
const int INF = <<;
const db eps = 1e-;
const ll mod = 1e9+;
struct Edge{
int u, v, next, val;
}edge[maxn*], edge1[maxn*];
int n, m;
int head[maxn], head1[maxn];
int father[maxn], dis[maxn], tol, toll, LCA[maxn];
bool visit[maxn];
int cnt;
//LCA[]最近公共祖先 void init() {
cnt=;
memset(head, -, sizeof(head));
memset(head1, -, sizeof(head1));
tol=toll=;
memset(visit, , sizeof(visit));
memset(LCA, , sizeof(LCA));
memset(dis, , sizeof(dis));
}
void add(int u, int v, int w) {
edge[tol].u=u, edge[tol].v=v,edge[tol].val=w;
edge[tol].next=head[u];
head[u]=tol++;
}
void add1(int u, int v, int w) {
edge1[toll].u=u, edge1[toll].v=v,edge1[toll].val=w;
edge1[toll].next=head1[u];
head1[u]=toll++;
}
int Find(int x) {
if (x!=father[x]) father[x]=Find(father[x]);
return father[x];
}
void tarjan(int u) {
visit[u]=;
father[u]=u;
for (int j=head1[u]; ~j; j=edge1[j].next) {
int v=edge1[j].v;
if (visit[v]) LCA[edge1[j].val]=Find(v);
}
for (int j=head[u]; ~j; j=edge[j].next) {
int v=edge[j].v;
if (!visit[v]) {
dis[v]=dis[u]+edge[j].val;
tarjan(v);
father[v]=u;
//cout<<dis[v]<<endl;
}
}
}
void solve() {
init();
scanf("%d%d", &n, &m);
for (int i=; i<n; i++) {
int a, b, c; scanf("%d%d%d", &a, &b, &c);
add(a, b, c); add(b, a, c);
}
for (int i=; i<m; i++) {
int a, b; scanf("%d%d", &a, &b);
add1(a, b, i); add1(b, a, i);
}
dis[]=;
tarjan();
for(int i=;i<toll;i+=){
printf("%d\n",LCA[edge1[i].val]);
}
}
int main() {
int t = ;
//freopen("in.txt","r",stdin);
// freopen("gcd.out","w",stdout);
scanf("%d", &t);
while(t--)
solve();
return ;
}
倍增版本
/* gyt
Live up to every day */
#pragma comment(linker,"/STACK:1024000000,1024000000")
#include<cstdio>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#include<stack>
#include<cstring>
#include<queue>
#include<set>
#include<string>
#include<map>
#include <time.h>
#define PI acos(-1)
using namespace std;
typedef long long ll;
typedef double db;
const int maxn = 1e5+;
const ll maxm = 1e7;
const ll base = ;
const int INF = <<;
const db eps = 1e-;
const ll mod = 1e9+;
struct Edge{
int u, v, next, val;
}edge[maxn*];
int head[maxn], dis[maxn];
int f[maxn], deep[maxn], p[maxn][];
int n, m, cnt; void init() {
int i,j;
for(j=;(<<j)<=n;j++)
for(i=;i<=n;i++)
p[i][j]=-;
for(i=;i<=n;i++)p[i][]=f[i];
for(j=;(<<j)<=n;j++)
for(i=;i<=n;i++)
if(p[i][j-]!=-)
p[i][j]=p[p[i][j-]][j-];//i的第2^j祖先就是i的第2^(j-1)祖先的第2^(j-1)祖先
}
void add(int u, int v, int w) {
edge[cnt].u=u, edge[cnt].v=v;
edge[cnt].next=head[u], edge[cnt].val=w;
head[u]=cnt++;
}
void dfs(int u, int pre, int t) {
deep[u]=t;
f[u]=pre;
for (int i=head[u]; ~i; i=edge[i].next) {
int v=edge[i].v;
if (v!=pre) {
dis[v]=dis[u]+edge[i].val;
dfs(v, u, t+);
}
}
}
int lca(int a, int b) {
int i, j;
if (deep[a]<deep[b]) swap(a, b);
for ( i=; (<<i)<=deep[a]; i++);
i--;
for ( j=i; j>=; j--) {
if (deep[a]-(<<j)>=deep[b]) a=p[a][j];
}
if (a==b) return a;
for ( j=i; j>=; j--) {
if (p[a][j]!=- && p[a][j]!=p[b][j]) {
a=p[a][j], b=p[b][j];
}
}
return f[a];
}
void solve() {
scanf("%d%d", &n, &m);
cnt=;
memset(head, -, sizeof(head));
for (int i=; i<n-; i++) {
int a, b, c; scanf("%d%d%d", &a, &b, &c);
add(a, b, c); add(b, a, c);
}
dis[]=;
dfs(, -, );
init();
for (int i=; i<m; i++) {
int a, b; scanf("%d%d", &a, &b);
int ans=dis[a]+dis[b]-*dis[lca(a, b)];
printf("%d\n", ans);
}
}
int main() {
int t = ;
// freopen("in.txt","r",stdin);
// freopen("gcd.out","w",stdout);
scanf("%d", &t);
while(t--)
solve();
return ;
}
LCA(最近公共祖先)模板的更多相关文章
- LCA最近公共祖先模板(求树上任意两个节点的最短距离 || 求两个点的路进(有且只有唯一的一条))
原理可以参考大神 LCA_Tarjan (离线) TarjanTarjan 算法求 LCA 的时间复杂度为 O(n+q) ,是一种离线算法,要用到并查集.(注:这里的复杂度其实应该不是 O(n+q) ...
- LCA最近公共祖先模板代码
vector模拟邻接表: #include<iostream> #include<cstdio> #include<cstring> #include<cma ...
- lca最短公共祖先模板(hdu2586)
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2586 #include<iostream> #include<cstdio> ...
- LCA(最近公共祖先)之倍增算法
概述 对于有根树T的两个结点u.v,最近公共祖先LCA(T,u,v)表示一个结点x,满足x是u.v的祖先且x的深度尽可能大. 如图,3和5的最近公共祖先是1,5和2的最近公共祖先是4 在本篇中我们先介 ...
- lca 最近公共祖先
http://poj.org/problem?id=1330 #include<cstdio> #include<cstring> #include<algorithm& ...
- Tarjan算法应用 (割点/桥/缩点/强连通分量/双连通分量/LCA(最近公共祖先)问题)(转载)
Tarjan算法应用 (割点/桥/缩点/强连通分量/双连通分量/LCA(最近公共祖先)问题)(转载) 转载自:http://hi.baidu.com/lydrainbowcat/blog/item/2 ...
- CodeVs.1036 商务旅行 ( LCA 最近公共祖先 )
CodeVs.1036 商务旅行 ( LCA 最近公共祖先 ) 题意分析 某首都城市的商人要经常到各城镇去做生意,他们按自己的路线去做,目的是为了更好的节约时间. 假设有N个城镇,首都编号为1,商人从 ...
- LCA近期公共祖先
LCA近期公共祖先 该分析转之:http://kmplayer.iteye.com/blog/604518 1,并查集+dfs 对整个树进行深度优先遍历.并在遍历的过程中不断地把一些眼下可能查询到的而 ...
- LCA 近期公共祖先 小结
LCA 近期公共祖先 小结 以poj 1330为例.对LCA的3种经常使用的算法进行介绍,分别为 1. 离线tarjan 2. 基于倍增法的LCA 3. 基于RMQ的LCA 1. 离线tarjan / ...
随机推荐
- PHP 获取 IE浏览器版本号
function getIEBrowserVersion(){ $agent = strtolower($_SERVER['HTTP_USER_AGENT']); if(strpos($agent, ...
- serv-U使用
该软件是设置ftp服务器的 可以百度查询ftp服务器安装攻略,如 https://jingyan.baidu.com/article/cb5d6105c00bba005c2fe0ca.html 问题: ...
- LibreOJ 6277. 数列分块入门 2
题目链接:https://loj.ac/problem/6278 参考博客:https://blog.csdn.net/qq_36038511/article/details/79725027 这题我 ...
- ansible1
前期工作: 第一步:下载epel源 wget -O /etc/yum.repos.d/epel.repo http://mirrors.aliyun.com/repo/epel-7.repo 第二步: ...
- java里面获取map的key和value的方法
获取map的key和value的方法分为两种形式: map.keySet():先获取map的key,然后根据key获取对应的value: map..entrySet():同时查询map的key和val ...
- POJ-1321.棋盘问题.(回溯)
做完题之后看了网上的一些题解但是发现他们的解释大部分都是错误的,所以就自己写了一下,笔者能力也有限,有错误之处大家多多指正. 第一次看题的时候以为就是简单的八皇后,但是写了之后发现存在很多问题,比如需 ...
- String int 相互转换
String->int: int i = Integer.parseInt(s) int->String: String s = Integer.toString(i)
- Mac 动态库加载不上
OC xcode can't found xxx.dylib 1 targer- build phase :link binary with library添加动态库 注意不要将后边的选项选成opti ...
- cloud配置中心遇到的坑
https://blog.csdn.net/z960339491/article/details/80593982分布式配置中心为什么要有用分布式配置中心这玩意儿?现在这微服务大军已经覆盖了各种大小型 ...
- SA9 collections
[定义] 表示object的集合 generic class:可以用于多种object, 抽象类的具体实现: [ArrayList] 动态添加,只能加Non-primitive type,要初始化长 ...