HDU 4547 - CD操作 （LCA）

CD操作

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 2726    Accepted Submission(s): 752

Problem Description

　　在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

23 1B AC AB C3 2B AC BA CC A

 

Sample Output

212

 

Source

2013金山西山居创意游戏程序挑战赛——初赛（1） 

 

注意一定要a变成b，不能b变成a。

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <string>
#include <algorithm>
#include <map>
using namespace std;
#define LL long long
#define lt x<<1
#define rt x<<1|1
const int maxn = 100000+111;
int head[maxn],to[maxn<<1],nxt[maxn<<1],cnt=0;
int fa[maxn][30],d[maxn];
map<string,int >mp;
int n,m;
void add(int x,int y)
{
    to[cnt]=y;
    nxt[cnt]=head[x];
    head[x]=cnt++;
}
void dfs(int u,int pre)
{
    d[u]=d[pre]+1;
    for(int i=head[u];~i;i=nxt[i]){
        int v=to[i];
        if(v==pre) continue;
        dfs(v,u);
        fa[v][0]=u;
    }
}
void init()
{
 
    for(int i=1;(1<<i)<=n;i++){
       for(int j=1;j<=n;j++)
        fa[j][i]=fa[fa[j][i-1]][i-1];
    }
}

int query(int x,int y)
{
    if(x==y) return 0;
    int a=x,b=y;
    if(d[x]<d[y]) swap(x, y);
    int dis=d[x]-d[y];
    int c=0;
    while(dis){
        if(dis&1) x=fa[x][c];
        dis/=2;
        c++;
    }
    if(x!=y){
        for(int i = (int)log2(n);i>=0;i--){
        if(fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
        }
        x=fa[x][0];
    }
    if(a==x){
        return 1;
    }
    if(x==b){
        return d[a]-d[x];
    }else
        return d[a]-d[x]+1;


}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        cnt=0;
        memset(head,-1,sizeof head);
        memset(d,0,sizeof d);
        mp.clear();
        scanf("%d %d",&n,&m);
        int temp=0;
        int k[maxn];
        memset(k,0,sizeof k);
        for(int i=1;i<n;i++){
            char s1[44],s2[44];
            scanf("%s %s",s1,s2);
            if(mp[s1]==0) mp[s1]=++temp;
            if(mp[s2]==0) mp[s2]=++temp;
            k[mp[s1]]++;
            add(mp[s1],mp[s2]);
            add(mp[s2],mp[s1]);
        }
        for(int i=1;i<=n;i++){
            if(k[i]==0){
                d[0]=0;
                dfs(i,0);
                break;
            }
        }
        init();
        for(int i=1;i<=m;i++){
            char s1[44],s2[44];
            scanf("%s %s",s1,s2);
            printf("%d\n",query(mp[s1],mp[s2]));
        }

    }

    return 0;
}