HDU-4547 CD操作 （离线Tarjan）

CD操作

　在Windows下我们可以通过cmd运行DOS的部分功能，其中CD是一条很有意思的命令，通过CD操作，我们可以改变当前目录。 
　　这里我们简化一下问题，假设只有一个根目录，CD操作也只有两种方式： 
　　 
　　1. CD 当前目录名\...\目标目录名 (中间可以包含若干目录，保证目标目录通过绝对路径可达) 
　　2. CD .. (返回当前目录的上级目录) 
　　 
　　现在给出当前目录和一个目标目录，请问最少需要几次CD操作才能将当前目录变成目标目录？

Input

输入数据第一行包含一个整数T(T<=20)，表示样例个数； 
每个样例首先一行是两个整数N和M(1<=N,M<=100000)，表示有N个目录和M个询问； 
接下来N-1行每行两个目录名A B(目录名是只含有数字或字母，长度小于40的字符串)，表示A的父目录是B。 
最后M行每行两个目录名A B，表示询问将当前目录从A变成B最少要多少次CD操作。 
数据保证合法，一定存在一个根目录，每个目录都能从根目录访问到。

Output

请输出每次询问的结果，每个查询的输出占一行。

Sample Input

2
3 1
B A
C A
B C

3 2
B A
C B
A C
C A

Sample Output

2
1
2

题目大意：一共有三种情况：1.u是v的祖先1次就到， 2.v是u的祖先 v到u的距离，3.否则u到公共祖先的距离+1

代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<map>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
const int N=100000+20;
int first[N],first1[N],fa[N],d[N],vis[N],vis1[N],isroot[N],ans[N];
int tot,tott,root,n,m;
map<string,int>mp;
struct node
{
    int v,next;
}e[N<<1];
struct query
{
    int v,next,id,flag;
}q[N<<1];
struct poi
{
    int u,v;
}qq[N<<1];
void adde(int u,int v)
{
    e[tot].v=v;
    e[tot].next=first[u];
    first[u]=tot++;
}
void addee(int u,int v,int id,int flag)
{
    q[tott].v=v;
    q[tott].id=id;
    q[tott].flag=flag;
    q[tott].next=first1[u];
    first1[u]=tott++;
}
void init()
{
    mem(first,-1);
    mem(first1,-1);
    mem(d,0);
    mem(vis,0);
    mem(vis1,0);
    mem(ans,0);
    mem(isroot,0);
    tot=tott=0;
    mp.clear();
    for(int i=1;i<=n;i++) fa[i]=i;
}
int find(int x)
{
    if(fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
void dfs(int u,int len)
{
    d[u]=len;
    vis1[u]=1;
    for(int i=first[u];~i;i=e[i].next)
    {
        int v=e[i].v;
        if(!vis1[v])
            dfs(v,len+1);
    }
}
void LCA(int u,int pre)
{
    for(int i=first[u];~i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==pre) continue;
        LCA(v,u);
        fa[v]=u;
    }
    vis[u]=1;
    for(int i=first1[u];~i;i=q[i].next)
    {
        int v=q[i].v;
        if(vis[v])
        {
            int id=q[i].id,flag=q[i].flag;
            int u1=u,v1=v;
            if(flag==2)
                swap(u1,v1);
            if(u1==find(v1)) ans[id]=1;
            else if(v1==find(u1)) ans[id]=d[u1]-d[find(v)];
            else ans[id]=d[u1]-d[find(v)]+1;
        }
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        string s1,s2;
        scanf("%d%d",&n,&m);
        init();
        int u,v,cnt=0;
        for(int i=1;i<n;i++)
        {
            cin>>s1>>s2;
            if(!mp[s1]) mp[s1]=++cnt;
            if(!mp[s2]) mp[s2]=++cnt;
            u=mp[s2],v=mp[s1];
            adde(u,v);
            adde(v,u);
            isroot[v]=1;
        }
        for(int i=1;i<=n;i++)
        if(!isroot[i]){root=i;break;}
        dfs(root,0);
        for(int i=1;i<=m;i++)
        {
            cin>>s1>>s2;
            u=mp[s1],v=mp[s2];
            qq[i].u=u,qq[i].v=v;
            addee(u,v,i,1);
            addee(v,u,i,2);
        }
        LCA(root,root);
        for(int i=1;i<=m;i++)
        {
            u=qq[i].u,v=qq[i].v;
            if(u==v) printf("0\n");
            else
                printf("%d\n",ans[i]);
        }
    }
    return 0;
}