P3379 【模板】最近公共祖先(LCA)
题目描述
如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。
输入输出格式
输入格式:
第一行包含三个正整数N、M、S,分别表示树的结点个数、询问的个数和树根结点的序号。
接下来N-1行每行包含两个正整数x、y,表示x结点和y结点之间有一条直接连接的边(数据保证可以构成树)。
接下来M行每行包含两个正整数a、b,表示询问a结点和b结点的最近公共祖先。
输出格式:
输出包含M行,每行包含一个正整数,依次为每一个询问的结果。
输入输出样例
输入样例#1: 复制
5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5
输出样例#1: 复制
4
4
1
4
4
说明
时空限制:1000ms,128M
数据规模:
对于30%的数据:N<=10,M<=10
对于70%的数据:N<=10000,M<=10000
对于100%的数据:N<=500000,M<=500000
样例说明:
该树结构如下:
第一次询问:2、4的最近公共祖先,故为4。
第二次询问:3、2的最近公共祖先,故为4。
第三次询问:3、5的最近公共祖先,故为1。
第四次询问:1、2的最近公共祖先,故为4。
第五次询问:4、5的最近公共祖先,故为4。
故输出依次为4、4、1、4、4。
http://www.cnblogs.com/JVxie/p/4854719.html
#include <bits/stdc++.h>
using namespace std;
const int N = 500007;
const int M = 1e6 + 7;
int f[N];
int root(int u) { //并查集
return u == f[u] ? u :f[u] = root(f[u]);
}
int head[N], en;
int head_[N], next_[M], link[M], tot; //用来存询问的邻接表
int ans[N], vis[N];
struct Edge {
int v, to;
} e[M];
void addedge(int u, int v) {
e[++en].v = v;
e[en].to = head[u];
head[u] = en;
e[++en].v = u;
e[en].to = head[v];
head[v] = en;
}
void add_(int u, int v) {
link[++tot] = v;
next_[tot] = head_[u];
head_[u] = tot;
link[++tot] = u;
next_[tot] = head_[v];
head_[v] = tot;
}
void init(int n) {
memset(head, -1, sizeof(head));
memset(head_, -1, sizeof(head_));
memset(vis, 0, sizeof(vis));
en = tot = 0;
for (int i = 1; i <= n; ++i) f[i] = i;//并查集初始化
}
void dfs(int u, int father) { //
for (int i = head[u]; ~i; i = e[i].to) {
int v = e[i].v;
if (v != father) {
dfs(v, u);
f[root(v)] = u; //合并v到u上
vis[v] = 1;//表示v节点已经处理完
}
}
for (int i = head_[u]; ~i; i = next_[i]) {
int v = link[i];
if (vis[v]) { //v已经访问过,则u和v的lca为v所合并到的节点
ans[(i+1)/2] = root(v); //因为询问是按邻接表有顺序存的,所以答案也是对应的
} //除以2是因为双向边
}
}
int main() {
int n, m, s;
scanf ("%d %d %d", &n, &m, &s);
int u, v;
init(n);
for (int i = 1; i < n; ++i) {
scanf ("%d %d", &u, &v);
addedge (u, v);
}
for (int i = 1; i <= m; ++i) {
scanf ("%d %d", &u, &v);
add_(u, v); //邻接表存询问,以便处理有关系的点
}
dfs(s, s);
for (int i = 1; i <= m; ++i) { //按顺序输出答案
printf ("%d\n", ans[i]);
}
}