hdu4547 CD操作（lca+map）

就是求lca。。。要注意如果A=B的话是需要0步。字符串映射可以采用map。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <map>
using namespace std;
#define N 100010
#define ll long long
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,dep[N],fa[N][20],rt,h[N],num=0,Log[N];
bool f[N];
struct edge{
    int to,next;
}data[N];
map<string,int>mp;
string s1,s2;
inline void add(int x,int y){
    data[++num].to=y;data[num].next=h[x];h[x]=num;
}
void dfs(int x){
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;dep[y]=dep[x]+1;dfs(y);
    }
}
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=0;i<=Log[d];++i)
        if(d>>i&1) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int main(){
//  freopen("a.in","r",stdin);
    int tst=read();Log[0]=-1;
    for(int i=1;i<=N-10;++i) Log[i]=Log[i>>1]+1;
    while(tst--){
        n=read();m=read();int tot=0;memset(f,0,sizeof(f));
        memset(fa,0,sizeof(fa));num=0;memset(h,0,sizeof(h));mp.clear();
        for(int i=1;i<n;++i){
            cin>>s1>>s2;
            if(!mp[s1]) mp[s1]=++tot;
            if(!mp[s2]) mp[s2]=++tot;
            fa[mp[s1]][0]=mp[s2];f[mp[s1]]=1;add(mp[s2],mp[s1]);
        }for(int i=1;i<=n;++i) if(!f[i]) rt=i;
        dep[rt]=0;dfs(rt);
        while(m--){
            cin>>s1>>s2;
            int x=mp[s1],y=mp[s2],t=lca(x,y);
            if(x==y) puts("0");
            else if(x==t) puts("1");
            else if(y==t) printf("%d\n",dep[x]-dep[y]);
            else printf("%d\n",dep[x]-dep[t]+1);
        }
    }return 0;
}