思路:
此做法不是很难,运用了我所熟知的并查集。但由于我把二维数组打成了一维数组,仍然花了点时间才交上去。
设 f a i fa_i fai 为 i i i 的祖先, v e c t o r < i n t > e i j vector<int>e_{ij} vector<int>eij 维护整棵树, v i s i vis_i visi 判断每个节点是否搜过, p i j p_{ij} pij 记录询问的两条边和询问编号, a n s i ans_i ansi 存储相应编号的答案。
思路核心:进回退
这个做法是边搜边查,利用深搜遍历每个点,遍历到 u 时搜索它的子节点(进)。
搜完之后将 u 的那个子节点的祖宗(并查集的)设为 u(回)。
然后搜索下一个子节点。当所有子节点都搜完之后,退回到 u 的上一个节点(退)。
注意:回时候的找祖宗对 u 同样适用。
另附:Tarjan 算法属于离线算法,即它的每一步操作得出的东西都属于答案的一部分。
代码:
核心代码:
void tarjan(int f){
vis[f]=1;
for(auto i:e[f]){
if(!vis[i]){
tarjan(i);
fa[i]=f;
}
}
for(auto q:p[f]){
int v=q.first,u=q.second;
if(vis[v])ans[u]=find(v);
}
}
全代码:
#include<bits/stdc++.h>
using namespace std;
const int N=5e5+100;
int n,m,s;
int fa[N],ans[N],vis[N];
vector<int>e[N];
vector<pair<int,int>>p[N];
int find(int h){
if(fa[h]==h)return h;
return fa[h]=find(fa[h]);
}
void tarjan(int f){
vis[f]=1;
for(auto i:e[f]){
if(!vis[i]){
tarjan(i);
fa[i]=f;
}
}
for(auto q:p[f]){
int v=q.first,u=q.second;
if(vis[v]){
ans[u]=find(v);
}
}
}
int main(){
cin>>n>>m>>s;
for(int i=1;i<n;i++){
int a,b;
cin>>a>>b;
e[a].push_back(b);
e[b].push_back(a);
}
for(int i=1;i<=n;i++)fa[i]=i;
for(int i=1;i<=m;i++){
int a,b;
cin>>a>>b;
p[a].push_back({b,i});
p[b].push_back({a,i});
}
tarjan(s);
for(int i=1;i<=m;i++)cout<<ans[i]<<"\n";
return 0;
}