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操作才能将当前目录变成目标目录?
这里我们简化一下问题,假设只有一个根目录,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操作。
数据保证合法,一定存在一个根目录,每个目录都能从根目录访问到。
每个样例首先一行是两个整数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
注意一定要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;
}