您的位置:首页 > 其它

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

2018-01-17 11:55 525 查看
题目:最近公共祖先(LCA)

思路:tarjan。

代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <cstring>
#include <map>
using namespace std;

#define maxn 500000

int n,m,rt;

vector<int> a[maxn+5];

struct V{
int y,id;
V(){}
V(const int yy,const int idd){
y=yy,id=idd;
}
};
vector<V> q[maxn+5];

int fa[maxn+5]={0};

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

int lca[maxn+5]={0};

void tarjan(int x,int t){
for(int i=0;i<a[x].size();i++){
int u=a[x][i];
if(u==t) continue;
tarjan(u,x);
fa[u]=x;
}

for(int i=0;i<q[x].size();i++){
V y=q[x][i];
int f=find(y.y);
if(f!=y.y){
lca[y.id]=f;
}
}
}

int main() {

scanf("%d%d%d",&n,&m,&rt);

for(int i=1;i<n;i++){
int x,y;
scanf("%d%d",&x,&y);
a[x].push_back(y);
a[y].push_back(x);
}

for(int i=1;i<=m;i++){
int x,y;
scanf("%d%d",&x,&y);
q[x].push_back(V(y,i));
q[y].push_back(V(x,i));
}

tarjan(rt,-1);

for(int i=1;i<=m;i++){
printf("%d\n",lca[i]);
}

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息