HDU 4547

题意: 单向路径,从根目录可以通过一次操作到达子目录,从子目录到达上一级目录需要一次操作,问从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
*/


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值