9018:1892: 最近公共祖先用Tarjan来做
题目描述
编写一个程序,查找一棵含有n个节点的树中m组两个不同节点的最近公共祖先。
输入
第一行两个整数n和q,为树上节点的数目和查询数。节点标号为整数 1,2,...,n。之后n-1行每一行包含一对整数,代表边——第一个整数是第二个整数的父节点。请注意,一个具有n个节点的树有n-1条边。之后q行查询,每行两个整数,表示要查询的两个节点的标号。(输入保证有且只有一棵树)
输出
q行,每行一个整数,为此次查询的最近公共祖先的标号
样例输入
5 6
2 5
2 3
5 1
1 4
1 3
2 3
5 4
1 5
4 5
5 4
样例输出
2
2
5
5
5
5
提示
2<=n,m<=5*10^5
代码
#include<iostream>
#include<cstdlib>
#include<cstdio>
# define INF 500010
struct edge{int nx,t;}e[INF*3+1];
int h[INF],q[INF],count,ans[INF],f[INF],r[INF];
inline void ins(int*h,int x,int y){e[++count]=(edge){h[x],y};h[x]=count;}
int gf(int k){return f[k]?f[k]=gf(f[k]):k;}
void tj(int x){
int i;
for(i=q[x];i;i=e[i].nx)
if(ans[e[i].t])ans[e[i].t]=gf(ans[e[i].t]);//最后一个ans
else ans[e[i].t]=x;
for(i=h[x];i;i=e[i].nx)tj(e[i].t),f[e[i].t]=x;
}
int main(){
int m,n;
scanf("%d%d",&n,&m);
int i,x,y;
for(i=1;i<n;++i){scanf("%d",&x);scanf("%d",&y);r[y]=1;ins(h,x,y);}//++i,怎样输入
for(i=0;i<m;++i){scanf("%d",&x);scanf("%d",&y);ins(q,x,i);ins(q,y,i);}//++i,怎样输入
for(i=1;i<=n;++i){if(!r[i]){tj(i);break;}}//!,++i
for(i=0;i<m;++i)printf("%d\n",ans[i]);// ++i
}
总结:一定要仔细