首页 > 代码库 > hdu 1269 迷宫城堡 (tarjan算法学习)

hdu 1269 迷宫城堡 (tarjan算法学习)

迷宫城堡

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 7056    Accepted Submission(s): 3137


Problem Description
为了训练小希的方向感,Gardon建立了一座大城堡,里面有N个房间(N<=10000)和M条通道(M<=100000),每个通道都是单向的,就是说若称某通道连通了A房间和B房间,只说明可以通过这个通道由A房间到达B房间,但并不说明通过它可以由B房间到达A房间。Gardon需要请你写个程序确认一下是否任意两个房间都是相互连通的,即:对于任意的i和j,至少存在一条路径可以从房间i到房间j,也存在一条路径可以从房间j到房间i。
 

Input
输入包含多组数据,输入的第一行有两个数:N和M,接下来的M行每行有两个数a和b,表示了一条通道可以从A房间来到B房间。文件最后以两个0结束。
 

Output
对于输入的每组数据,如果任意两个房间都是相互连接的,输出"Yes",否则输出"No"。
 

Sample Input
3 3 1 2 2 3 3 1 3 3 1 2 2 3 3 2 0 0
 

Sample Output
Yes No
 

Author
Gardon
 

Source
HDU 2006-4 Programming Contest

模拟栈写的tarjan算法:

#include"stdio.h"
#include"string.h"
#include"queue"
#include"vector"
#include"stack"
#include"algorithm"
using namespace std;
#define N 10005
#define min(a,b) (a<b?a:b)
int n,t,stop,bcnt,index;
int dfn[N],instack[N],head[N];
int low[N],stap[N],belong[N];
int ans;    
struct node
{
    int u,v,next;
}e[N*10];
void add(int u,int v) //邻接表存边
{
    e[t].u=u;
    e[t].v=v;
    e[t].next=head[u];
    head[u]=t++;
}
void tarjan(int i)
{
    int j,t;
    dfn[i]=low[i]=++index;   //Low(u)为u或u的子树能够追溯到的最早的栈中节点的次序号
    instack[i]=true;         //模拟栈的存储方式,代表节点i入栈(标记)
    stap[++stop]=i;          //stop指向栈顶元素,stap记录栈内元素
    for(t=head[i];t!=-1;t=e[t].next)
    {
        j=e[t].v;
        if(!dfn[j])
        {
            tarjan(j);
            low[i]=min(low[i],low[j]);
        }
        else if(instack[j])
            low[i]=min(low[i],dfn[j]);
    }
    if(dfn[i]==low[i])
    {
        bcnt++;
        do
        {
            j=stap[stop--];   //取栈顶元素
            instack[j]=false;    //出栈
            belong[j]=bcnt;      //记录该强连通分量的代号
        }
        while(j!=i);
        ans++;
    }
}
void solve()
{
    int i;
    stop=bcnt=index=0;
    memset(dfn,0,sizeof(dfn));    //定义DFN(u)为节点u搜索的次序编号(时间戳)
    for(i=1;i<=n;i++)       
    {
        if(!dfn[i])
            tarjan(i);
    }
}
int main()
{
    int i,u,v,m;
    while(scanf("%d%d",&n,&m),n||m)
    {
        memset(head,-1,sizeof(head));
        t=0;
        ans=0;
        for(i=0;i<m;i++)
        {
            scanf("%d%d",&u,&v);
            add(u,v);
        }

        solve();
        if(ans==1)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}

一般STL容器栈stack:

#include"stdio.h"
#include"string.h"
#include"queue"
#include"vector"
#include"stack"
#include"algorithm"
using namespace std;
#define N 10005
#define min(a,b) (a<b?a:b)
int t,n,index,ans;
int head[N],instack[N];
int dfn[N],low[N];
struct node
{
    int u,v,next;
}e[N*10];
void add(int u,int v)
{
    e[t].u=u;
    e[t].v=v;
    e[t].next=head[u];
    head[u]=t++;
}
void tarjan(int u)
{
    int i,v;
    stack<int>s;
    s.push(u);
    dfn[u]=low[u]=index++;
    instack[u]=true;
    for(i=head[u];i!=-1;i=e[i].next)
    {
        v=e[i].v;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(instack[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(low[u]==dfn[u])
    {
        while(u!=v)
        {
            v=s.top();
            s.pop();
        }
        ans++;
    }
}
void solve()
{
    int i;
    memset(dfn,0,sizeof(dfn));
    index=1;
    for(i=1;i<=n;i++)
    {
        if(!dfn[i])
            tarjan(i);
    }
}
int main()
{
    int u,v,m;
    while(scanf("%d%d",&n,&m),n||m)
    {
        memset(head,-1,sizeof(head));
        ans=t=0;
        while(m--)
        {
            scanf("%d%d",&u,&v);
            add(u,v);
        }
        solve();
        if(ans==1)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}