lca最近公共祖先(模板)
洛谷上的lca模板题——传送门
1.tarjan求lca
学了求lca的tarjan算法(离线),在洛谷上做模板题,结果后三个点超时。
又把询问改成链式前向星,才ok。
这个博客,tarjan分析的很详细。
附代码——
#include <cstdio>
#include <cstring> const int maxn = ; int n, m, cnt, s, cns;
int x, y, z[maxn];//z是x和y的lca
int f[maxn], head[maxn], from[maxn];
bool vis[maxn];
struct node
{
int to, next;
}e[ * maxn];
struct Node
{
int to, next, num;
}q[ * maxn]; inline int read()//读入优化
{
int x = , f = ;
char ch = getchar();
while(ch < '' || ch > '')
{
if(ch == '-') f = -;
ch = getchar();
}
while(ch >= '' && ch <= '')
{
x = x * + ch - '';
ch = getchar();
}
return x * f;
} inline void ask(int u, int v, int i)//储存待询问的结构体,也是链式前向星优化
{
q[cns].num = i;//num表示第几次询问
q[cns].to = v;
q[cns].next = from[u];
from[u] = cns++;
} inline void add(int u, int v)//
{
e[cnt].to = v;
e[cnt].next = head[u];
head[u] = cnt++;
} inline int find(int a)
{
return a == f[a] ? a : f[a] = find(f[a]);//路径压缩优化
} /*inline void Union(int a, int b)
{
int fx = find(a), fy = find(b);
if(fx == fy) return;
f[fy] = fx;
}*/ inline void tarjan(int k)
{
int i, j;
vis[k] = ;
f[k] = k;
for(i = head[k]; i != -; i = e[i].next)
if(!vis[e[i].to])
{
tarjan(e[i].to);
//Union(k, e[i].to);
f[e[i].to] = k;
}
for(i = from[k]; i != -; i = q[i].next)
if(vis[q[i].to] == )
z[q[i].num] = find(q[i].to);
} int main()
{
int i, j, u, v;
n = read();
m = read();
s = read();
memset(head, -, sizeof(head));
memset(from, -, sizeof(from));
for(i = ; i <= n - ; i++)
{
u = read();
v = read();
add(u, v);//注意添加两遍
add(v, u);
}
for(i = ; i <= m; i++)
{
x = read();
y = read();
ask(x, y, i);//两遍
ask(y, x, i);
}
tarjan(s);
for(i = ; i <= m; i++) printf("%d\n", z[i]);
return ;
}
进过培训,修改了代码
# include <iostream>
# include <cstdio>
# include <cstring>
# include <string>
# include <cmath>
# include <vector>
# include <map>
# include <queue>
# include <cstdlib>
# define MAXN
using namespace std; inline int get_num() {
int k = , f = ;
char c = getchar();
for(; !isdigit(c); c = getchar()) if(c == '-') f = -;
for(; isdigit(c); c = getchar()) k = k * + c - '';
return k * f;
} int n, m, s;
int fa[MAXN], qx[MAXN], qy[MAXN], ans[MAXN], f[MAXN];
vector <int> vec[MAXN], q[MAXN]; inline int find(int x)
{
return x == fa[x] ? x : fa[x] = find(fa[x]);
} inline void dfs(int u)
{
int i, v;
fa[u] = u;
for(i = ; i < vec[u].size(); i++)
{
v = vec[u][i];
if(f[u] != v) f[v] = u, dfs(v);
}
for(i = ; i < q[u].size(); i++)
if(f[v = u ^ qx[q[u][i]] ^ qy[q[u][i]]])
ans[q[u][i]] = find(v);
fa[u] = f[u];
} int main()
{
int i, x, y;
n = get_num();
m = get_num();
s = get_num();
for(i = ; i < n; i++)
{
x = get_num();
y = get_num();
vec[x].push_back(y);
vec[y].push_back(x);
}
for(i = ; i <= m; i++)
{
qx[i] = get_num();
qy[i] = get_num();
q[qx[i]].push_back(i);
q[qy[i]].push_back(i);
}
dfs(s);
for(i = ; i <= m; i++) printf("%d\n", ans[i]);
return ;
}
其实上面两个代码有些重复运算,请手动把求lca的过程放到dfs上面(也就是遍历到这个节点就求lca,而不是遍历完再求)
2.倍增求lca
下面是求lca的倍增算法(在线)。
1. DFS预处理出所有节点的深度和父节点
inline void dfs(int u)
{
int i;
for(i=head[u];i!=-;i=next[i])
{
if (!deep[to[i]])
{
deep[to[i]] = deep[u]+;
p[to[i]][] = u; //p[x][0]保存x的父节点为u;
dfs(to[i]);
}
}
}
dfs预处理
2. 初始各个点的2^j祖先是谁 ,其中 2^j (j =0...log(该点深度))倍祖先,1倍祖先就是父亲,2倍祖先是父亲的父亲......。
void init()
{
int i,j;
//p[i][j]表示i结点的第2^j祖先
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)祖先
}
初始化
3.从深度大的节点上升至深度小的节点同层,如果此时两节点相同直接返回此节点,即lca。
否则,利用倍增法找到最小深度的 p[a][j]!=p[b][j],此时他们的父亲p[a][0]即lca。
int lca(int a,int b)//最近公共祖先
{
int i,j;
if(deep[a]<deep[b])swap(a,b);
for(i=;(<<i)<=deep[a];i++);
i--;
//使a,b两点的深度相同
for(j=i;j>=;j--)
if(deep[a]-(<<j)>=deep[b])
a=p[a][j];
if(a==b)return a;
//倍增法,每次向上进深度2^j,找到最近公共祖先的子结点
for(j=i;j>=;j--)
{
if(p[a][j]!=-&&p[a][j]!=p[b][j])
{
a=p[a][j];
b=p[b][j];
}
}
return p[a][];
}
倍增求lca
最后是完整代码,为了节约时间,就没有把p数组初始化为-1.
#include <cstdio>
#include <cstring>
#include <iostream> const int maxn = ;
int n, m, cnt, s;
int next[ * maxn], to[ * maxn], head[ * maxn], deep[maxn], p[maxn][]; inline void add(int x, int y)
{
to[cnt] = y;
next[cnt] = head[x];
head[x] = cnt++;
} inline void dfs(int i)
{
int j;
for(j = head[i]; j != -; j = next[j])
if(!deep[to[j]])
{
deep[to[j]] = deep[i] + ;
p[to[j]][] = i;
dfs(to[j]);
}
} inline void init()
{
int i, j;
for(j = ; ( << j) <= n; j++)
for(i = ; i <= n; i++)
p[i][j] = p[p[i][j - ]][j - ];
} inline int lca(int a, int b)
{
int i, j;
if(deep[a] < deep[b]) std::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[b][j])
{
a = p[a][j];
b = p[b][j];
}
return p[a][];
} int main()
{
int i, j, x, y;
memset(head, -, sizeof(head));
scanf("%d %d %d", &n, &m, &s);
for(i = ; i <= n - ; i++)
{
scanf("%d %d", &x, &y);
add(x, y);
add(y, x);
}
deep[s] = ;
dfs(s);
init();
for(i = ; i <= m; i++)
{
scanf("%d %d", &x, &y);
printf("%d\n", lca(x, y));
}
return ;
}
经过培训,又改了改代码。
# include <iostream>
# include <cstdio>
# include <cstring>
# include <string>
# include <cmath>
# include <vector>
# include <map>
# include <queue>
# include <cstdlib>
# define MAXN
using namespace std; inline int get_num() {
int k = , f = ;
char c = getchar();
for(; !isdigit(c); c = getchar()) if(c == '-') f = -;
for(; isdigit(c); c = getchar()) k = k * + c - '';
return k * f;
} int n, m, s;
int f[MAXN][], deep[MAXN];
vector <int> vec[MAXN]; inline void dfs(int u)
{
int i, v;
deep[u] = deep[f[u][]] + ;
for(i = ; f[u][i]; i++) f[u][i + ] = f[f[u][i]][i];
for(i = ; i < vec[u].size(); i++)
{
v = vec[u][i];
if(!deep[v]) f[v][] = u, dfs(v);
}
} inline int lca(int x, int y)
{
int i;
if(deep[x] < deep[y]) swap(x, y);
for(i = ; i >= ; i--)
if(deep[f[x][i]] >= deep[y])
x = f[x][i];
if(x == y) return x;
for(i = ; i >= ; i--)
if(f[x][i] != f[y][i])
x = f[x][i], y = f[y][i];
return f[x][];
} int main()
{
int i, x, y;
n = get_num();
m = get_num();
s = get_num();
for(i = ; i < n; i++)
{
x = get_num();
y = get_num();
vec[x].push_back(y);
vec[y].push_back(x);
}
dfs(s);
for(i = ; i <= m; i++)
{
scanf("%d %d", &x, &y);
printf("%d\n", lca(x, y));
}
return ;
}
3.树剖法求lca
# include <iostream>
# include <cstdio>
# include <cstring>
# include <string>
# include <cmath>
# include <vector>
# include <map>
# include <queue>
# include <cstdlib>
# define MAXN
using namespace std; inline int get_num() {
int k = , f = ;
char c = getchar();
for(; !isdigit(c); c = getchar()) if(c == '-') f = -;
for(; isdigit(c); c = getchar()) k = k * + c - '';
return k * f;
} int n, m, s;
int f[MAXN], size[MAXN], top[MAXN], son[MAXN], deep[MAXN];
vector <int> vec[MAXN]; inline void dfs1(int u)
{
int i, v;
size[u] = ;
deep[u] = deep[f[u]] + ;
for(i = ; i < vec[u].size(); i++)
{
v = vec[u][i];
if(!deep[v])
{
f[v] = u;
dfs1(v);
size[u] += size[v];
if(size[son[u]] < size[v]) son[u] = v;
}
}
} inline void dfs2(int u, int tp)
{
int i, v;
top[u] = tp;
if(!son[u]) return;
dfs2(son[u], tp);
for(i = ; i < vec[u].size(); i++)
{
v = vec[u][i];
if(v != son[u] && v != f[u]) dfs2(v, v);
}
} inline int lca(int x, int y)
{
while(top[x] != top[y])
{
if(deep[top[x]] < deep[top[y]]) swap(x, y);
x = f[top[x]];
}
if(deep[x] > deep[y]) swap(x, y);
return x;
} int main()
{
int i, x, y;
n = get_num();
m = get_num();
s = get_num();
for(i = ; i < n; i++)
{
x = get_num();
y = get_num();
vec[x].push_back(y);
vec[y].push_back(x);
}
dfs1(s);
dfs2(s, s);
for(i = ; i <= m; i++)
{
x = get_num();
y = get_num();
printf("%d\n", lca(x, y));
}
return ;
}
lca最近公共祖先(模板)的更多相关文章
- LCA(最近公共祖先)模板
Tarjan版本 /* gyt Live up to every day */ #pragma comment(linker,"/STACK:1024000000,1024000000&qu ...
- 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 / ...
随机推荐
- 在Vue中通过自定义指令获取元素
vue.js 是数据绑定的框架,大部分情况下我们都不需要直接操作 DOM Element,但在某些时候,我们还是有获取DOM Element的需求的: 在 vue.js 中,获取某个DOM Eleme ...
- KoaHub.js -- 基于 Koa.js 平台的 Node.js web 快速开发框架之koahub
Installation $ npm install koahub Use with koa var app = require('koa')(); var router = require( ...
- iTunes Documents 文件共享指导手册
金田 p.p1 { margin: 0.0px 0.0px 0.0px 0.0px; font: 11.0px Helvetica; min-height: 13.0px } p.p2 { margi ...
- 在python&numpy中切片(slice)
在python&numpy中切片(slice) 上文说到了,词频的统计在数据挖掘中使用的频率很高,而切片的操作同样是如此.在从文本文件或数据库中读取数据后,需要对数据进行预处理的操作.此时就 ...
- idea调试SpringMvc, 出现:java.lang.ClassNotFoundException: org.springframework.web.context.ContextLoaderListener错误的解决办法
有时,使用idea开发SpringMvc发现调试时出现以下错误: 12-Mar-2017 12:08:02.345 严重 [RMI TCP Connection(2)-127.0.0.1] org.a ...
- windows phone 8.1 FlipView 实现照片自动浏览
FlipView 控件也是一个集合控件,不过它和ListView等控件不同,ListView控件是一次显示多个项,而FlipView则是每次只显示一个项.在windows phone上通过默认的左右滑 ...
- Butter Knife使用详解
Butter Knife Github地址: https://github.com/JakeWharton/butterknife 官方说明给出的解释是 Bind Android views and ...
- 编写PHP代码总结
1- 编写模块化代码 良好的PHP代码应该是模块化代码.PHP的面向对象的编程功能是一些特别强大的工 具,可以把你的应用程序分解成函数或方法.你应该尽可能多的从你的应用程序的服务器端分开前端的HTML ...
- MVC5 DB FIRST
跟着师父一直在做codefirst的开发,最近有个新需求,就是需要人家的数据库,然后来开发,现在出现问题了.整理如下 目前有个现成的我们之前的codefirst的工程代码,我记得师父说过,根据数据库生 ...
- centos6.5安装配置supervisor
1.下载并安装supervisor https://pypi.python.org/pypi/supervisor/3.2.0 .tar.gz cd supervisor- python setup. ...