LCA最近公共祖先(tarjan离线算法)

/**LCA算法：首先用CD ..回退到最近公共祖先，再一次性到达目的地。*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
#include <algorithm>
using namespace std;
#define mem(a)  memset(a,0,sizeof(a))
#define maxn 1502
#define M   50002
int indegree[maxn];// 用来查找root
int fa[maxn],vis[maxn],query[maxn][maxn];
vector<int> edge[maxn];
map<string,int>m;
int n;//图中节点的个数,编号1~n
struct Allquery{
    int a,b;
}p[M];
void init()
{
    m.clear();
    mem(indegree);
    mem(vis);
    mem(query);
    for(int i=1;i<maxn;i++)
        edge[i].clear();
}
int find_father(int i)
{
    if(i==fa[i])
        return i;
    find_father(fa[i]);//不压缩路径
}
void tarjan(int x)
{
    for(int i=1;i<=n;i++)
    if(vis[i]&&query[x][i])
    {
        query[x][i]=query[i][x]=find_father(i);
    }
    fa[x]=x;
    vis[x]=1;
    for(int i=0;i<edge[x].size();i++)
    {
        tarjan(edge[x][i]);
        fa[edge[x][i]]=x;
    }
}
int getDepth(int i,int f,int cnt)
{
    if(i==f)
        return cnt;
    getDepth(fa[i],f,cnt+1);
}
int main()
{
    int T,k,q;
    cin>>T;
    int i,a,b;
    char str1[42],str2[42];
    while(T--)
    {
        init();
        scanf("%d%d",&k,&q);
        n=0;
        for(int i=1;i<k;i++)
        {
            scanf("%s%s",str1,str2);
            if(m[str1]==0)m[str1]=++n;
            a=m[str1];
            if(m[str2]==0)m[str2]=++n;
            b=m[str2];
            edge[b].push_back(a);
            indegree[a]++;//a是孩子
        }
        //输入查询次数
        for(i=0;i<q;i++)
        {
            scanf("%s%s",str1,str2);
            p[i].a=m[str1];
            p[i].b=m[str2];
            query[p[i].a][p[i].b]=query[p[i].b][p[i].a]=1;
        }
        //找到root
        for(i=1;i<=n;i++)
        if(indegree[i]==0)
        {
            tarjan(i);break;
        }
        for(i=0;i<q;i++)
        {
            int f=query[p[i].a][p[i].b];
            int ans=getDepth(p[i].a,f,0);//求出a离f的距离
            if(p[i].b!=f) ans++;
            printf("%d\n",ans);
        }
    }
    return 0;
}