POJ1470 LCA（tarjan离线求最近公共祖先）

1. #include<cstdio>  
2. #include<cstring>  
3. const int N = 910;  
4.   
5. struct Edge{  
6.     int e,next;  
7. }edge[2*N];  
8.   
9. int n,m,e_num,head[N],vis[N],cnt[N];  
10. int x[N*N],y[N*N],f[N*N];  
11.   
12. void AddEdge(int a,int b){  
13.     edge[e_num].e=b; edge[e_num].next=head[a]; head[a]=e_num++;  
14.     edge[e_num].e=a; edge[e_num].next=head[b]; head[b]=e_num++;  
15. }  
16.   
17. int findx(int x){  
18.     if(f[x]!=x)return f[x]=findx(f[x]);  
19.     return f[x];  
20. }  
21.   
22. void tarjan(int k){  
23.     int i;  
24.     vis[k]=1;  
25.     f[k]=k;  
26.     for(i=1;i<=m;i++){//遍历处理刚才保存的 m 个询问  
27.         if(x[i]==k && vis[y[i]])cnt[findx(y[i])]++;  
28.         if(y[i]==k && vis[x[i]])cnt[findx(x[i])]++;  
29.     }  
30.     for(i=head[k];i!=-1;i=edge[i].next){  
31.         if(!vis[edge[i].e]){  
32.             tarjan(edge[i].e);  
33.             f[edge[i].e]=k;  
34.         }  
35.     }  
36. }  
37.   
38. int main()  
39. {  
40.     int i,t,id,a,flag[N];  
41.     char ch1[2],ch2[2],ch3[2];  
42.     while(~scanf("%d",&n))  
43.     {  
44.         e_num=0;  
45.         for(i=1;i<=n;i++){  
46.             flag[i]=vis[i]=cnt[i]=0;  
47.             head[i]=-1;  
48.         }  
49.         for(i=1;i<=n;i++){  
50.             scanf("%d%1s%1s%d%1s",&id,ch1,ch2,&t,ch3);  
51.             while(t--){  
52.                 scanf("%d",&a);  
53.                 flag[a]=1;  
54.                 AddEdge(id,a);  
55.             }  
56.         }  
57.         scanf("%d",&m);  
58.         for(i=1;i<=m;i++)  
59.             scanf("%1s%d%d%1s",ch1,&x[i],&y[i],ch2);  
60.   
61.         for(i=1;i<=n;i++)//注意，这里，根节点不一定是 1  
62.             if(flag[i]==0)break;  
63.   
64.         tarjan(i);  
65.   
66.         for(i=1;i<=n;i++)  
67.             if(cnt[i])printf("%d:%d\n",i,cnt[i]);  
68.     }  
69.     return 0;  
70. }