[hdu4547] CD操作

来源：互联网发布：抓老鼠知乎编辑：程序博客网时间：2024/06/17 06:42

题目链接

题目大意：给一棵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();    }}

0 0