POJ1330(Nearest Common Ancestors)

求一棵树中2个点的最近公共祖先。

我的做法：用并查集求出每个结点的深度，然后递归求最近公共祖先。

#include <stdio.h>
#define N 10001
int fa[N],p[N],d[N],n;
void make_set()
{
  for(int i=1;i<=n;i++)
  {
    p[i]=i;
    d[i]=0;
  }
}
int find_set(int i)
{
  int pi=p[i];
  if(pi!=i) p[i]=find_set(p[i]);
  if(pi!=p[i])  d[i]+=d[pi];
  return p[i];
}
void union_set(int i,int j)
{
  int pi=find_set(i),pj=find_set(j);
  fa[j]=i;
  p[pj]=pi;
  d[pj]=-d[j]+1+d[i];
}
int lca(int i,int j)
{
  if(i==j)  return i;
  if(d[i]>d[j]) return lca(fa[i],j);
  if(d[i]<d[j]) return lca(i,fa[j]);
  return lca(fa[i],fa[j]);
}
int main()
{
  int t,i,a,b;
  scanf("%d",&t);
  while(t--)
  {
    scanf("%d",&n);
    make_set();
    for(i=0;i<n-1;i++)
    {
      scanf("%d%d",&a,&b);
      union_set(a,b);
    }
    scanf("%d%d",&a,&b);
    printf("%d\n",lca(a,b));
  }
  return 0;
}