[hdu4547] CD操作

题目链接

题目大意：给一棵n(n<=100000)个节点的树，有两种移动方式：1.到父节点 2.到子树中任意一个节点
m次询问，求两个节点之间的最小次数

题解：需要hash……这里直接用的map。然后就是根据lca大力分类讨论了

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <unordered_map>
using namespace std;
const int M=100005;
int n,q,t,K,T;
int head[M],f[M][20],dep[M];

unordered_map <string,int> mp;

struct edge{
    int to,nex;
}e[500005];

void add(int i,int j)
{
    e[t].to=j;
    e[t].nex=head[i];
    head[i]=t++;
}

int lca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=0;i<=18;i++) if(dep[x]-dep[y]&(1<<i)) x=f[x][i];
    if(x==y) return x;
    for(int i=18;i>=0;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}

void del()
{
    for(int j=1;j<=18;j++)
    for(int i=1;i<=n;i++)
    f[i][j]=f[f[i][j-1]][j-1];
}

int dfs(int x)
{
    for(int i=head[x];i!=-1;i=e[i].nex)
    {
        int v=e[i].to;
        if(v!=f[x][0]){
            dep[v]=dep[x]+1;
            f[v][0]=x;
            dfs(v);
        }
    }
}

int ID(string x)//人民群众喜闻乐见的map
{
    if(mp.count(x)==0)
    return mp[x]=++K;
    return mp[x];
}

void work()
{
    string x,y;
    while(q--)
    {
        cin>>x>>y;
        int u=ID(x),v=ID(y);
        int z=lca(u,v);
        int d1=dep[u]-dep[z];//从u到z
        if(dep[v]!=dep[z]) d1++;//如果v不是u的祖先，需要从v-z
        if(u==v) d1=0;//数据有个大坑
        cout<<d1<<endl;
    }
}

void init()
{
    string x,y;
    memset(head,-1,sizeof(head));//……
    mp.clear();//……
    t=0;K=0;
    cin>>n>>q;
    for(int i=1;i<n;i++){
        cin>>x>>y;
        int v=ID(y),u=ID(x);
        add(u,v);add(v,u);
    }
    dep[1]=1;//个人爱好……
    dfs(1);//建树
    del();//倍增
}

int main()
{
    cin>>T;
    while(T--){
        init();
        work();
    }
}