题意: 单向路径,从根目录可以通过一次操作到达子目录,从子目录到达上一级目录需要一次操作,问从A目录到B目录需要多少次操作
分析:从出度为0的点开始搜索,具体看代码吧
#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<string>
#define N 100005
using namespace std;
map<string,int> mm;
int head_e[N],head_q[N],vis[N],ans[N*2],f[N],dis[N],cnt_e,cnt_q,cnt,in[N];
struct node {
int u,w,next,pre;
}e[N*2],q[N*2];
void addedge(int x,int y,int z){
e[cnt_e].u=x;e[cnt_e].w=z;e[cnt_e].next=head_e[y];head_e[y]=cnt_e++;
e[cnt_e].u=y;e[cnt_e].w=z;e[cnt_e].next=head_e[x];head_e[x]=cnt_e++;
}
void addquery(int x,int y,int z){
q[cnt_q].pre=1;q[cnt_q].u=y;q[cnt_q].w=z;q[cnt_q].next=head_q[x];head_q[x]=cnt_q++;
q[cnt_q].pre=0;q[cnt_q].u=x;q[cnt_q].w=z;q[cnt_q].next=head_q[y];head_q[y]=cnt_q++;
}
void init(){
memset(head_q,-1,sizeof(head_q));
memset(head_e,-1,sizeof(head_e));
memset(in,0,sizeof(in));
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
memset(ans,0,sizeof(ans)); // 一定要初始化,没有明白为什么
cnt_e=0;
cnt_q=0;
cnt=1;
}
int find(int x){
if(f[x]==x)
return x;
return f[x]=find(f[x]);
}
void tarjan(int k){
vis[k]=1;
f[k]=k;
int i,v;
for(i=head_q[k];i!=-1;i=q[i].next){
v=q[i].u;
if(vis[v]){
int rt=find(v);
if(!q[i].pre)
ans[q[i].w]=dis[v]-dis[rt]+(k==rt?0:1);
else
ans[q[i].w]=dis[k]-dis[rt]+(v==rt?0:1);
}
}
for(i=head_e[k];i!=-1;i=e[i].next){
v=e[i].u;
if(!vis[v]){
dis[v]=dis[k]+e[i].w;
tarjan(v);
f[v]=k;
}
}
}
int main()
{
string str1,str2;
int t,n,m,i,x,y;
scanf("%d",&t);
while(t--){
init();
scanf("%d%d",&n,&m);
mm.clear();
for(i=1;i<n;i++){
cin>>str1>>str2;
if(mm.find(str1)==mm.end())
mm[str1]=cnt++;
if(mm.find(str2)==mm.end())
mm[str2]=cnt++;
x=mm[str1];
y=mm[str2];
in[x]++;
addedge(x,y,1);
}
for(i=1;i<=m;i++){
cin>>str1>>str2;
addquery(mm[str1],mm[str2],i);
}
for(i=1;i<=n;i++){
if(!in[i]){ //从出度为0的点开始搜索
tarjan(i);
}
}
for(i=1;i<=m;i++){
printf("%d\n",ans[i]);
}
}
return 0;
}
/*
2
3 6
A B
C B
A C
C A
A B
B A
B C
C B
3 6
A B
C B
A C
C A
A A
B A
C C
B B
*/