求一棵树中2个点的最近公共祖先。
我的做法：用并查集求出每个结点的深度，然后递归求最近公共祖先。
View Code
1 #include <stdio.h> 2 #define N 10001 3 int fa[N],p[N],d[N],n; 4 void make_set() 5 { 6 for(int i=1;i<=n;i++) 7 { 8 p[i]=i; 9 d[i]=0; 10 } 11 } 12 int find_set(int i) 13 { 14 int pi=p[i]; 15 if(pi!=i) p[i]=find_set(p[i]); 16 if(pi!=p[i]) d[i]+=d[pi]; 17 return p[i]; 18 } 19 void union_set(int i,int j) 20 { 21 int pi=find_set(i),pj=find_set(j); 22 fa[j]=i; 23 p[pj]=pi; 24 d[pj]=-d[j]+1+d[i]; 25 } 26 int lca(int i,int j) 27 { 28 if(i==j) return i; 29 if(d[i]>d[j]) return lca(fa[i],j); 30 if(d[i]<d[j]) return lca(i,fa[j]); 31 return lca(fa[i],fa[j]); 32 } 33 int main() 34 { 35 int t,i,a,b; 36 scanf("%d",&t); 37 while(t--) 38 { 39 scanf("%d",&n); 40 make_set(); 41 for(i=0;i<n-1;i++) 42 { 43 scanf("%d%d",&a,&b); 44 union_set(a,b); 45 } 46 scanf("%d%d",&a,&b); 47 printf("%d\n",lca(a,b)); 48 } 49 return 0; 50 }