VIJOS 1321 魔塔

题意

在魔塔中有N个房间和M条道路，每条道路上有一个怪，它可以被一种特殊的武器消灭，而每个房间中也存在一种武器。现在知道第I个房间中的武器编号为I，小明（主人翁）初始在J房间，小明想知道哪些房间是他可以去的。
数据范围 1<=m<=50000,1<=a,b,J<=n<=50000
提示：m,n<50000不等于说数组可以只开到50000；输出前面无空格

分析

显然，一个房间在一个时刻能抵达，那么以后也可以
一个房间能被访问的条件是：1有相邻房间被访问 2相邻房间连着的道路可以通过
我们每访问一个房间i，都能使得一些新房间的条件得到满足，这些新房间就可能可以被访问

因此，我们在访问完房间i后
将i连向的房间检查一次能否访问
将怪物i守住的道路检查一次能否访问
这样就能保证每条路和每个节点都只被访问一次了
时间复杂度O(n+m)

代码

#include<cstdio>
#define fo(i,a,b) for(int i=a;i<=b;i++)
const int mn=50000+100,mm=50000+100;
int n,s,m,aa,bb,cc,be[mn],bc[mn],et,ct;
bool p[mn];
struct edge{
    int y,z,ne;
};
edge e[mm*2],c[mm];
void add_edge(int u,int v,int w){
    e[++et].y=v;
    e[et].z=w;
    e[et].ne=be[u];
    be[u]=et;
}
void add_edge2(int u,int v,int w){
    c[++ct].y=v;
    c[ct].z=w;
    c[ct].ne=bc[u];
    bc[u]=ct;
}
void dfs(int x){
    p[x]=1;
    int i=bc[x];
    while (i){
        int &y=c[i].y,&z=c[i].z;
        if (p[y]&&(!p[z]))
            dfs(z);
        if (p[z]&&(!p[y]))
            dfs(y);
        i=c[i].ne;
    }
    i=be[x];
    while (i){
        int &y=e[i].y,&z=e[i].z;//int &y=c[i].y,&z=c[i].z;
        if (p[z]&&(!p[y]))
            dfs(y);
        i=e[i].ne;
    }
}
int main(){
    scanf("%d%d%d",&n,&s,&m);
    fo(i,1,m){
        scanf("%d%d%d",&aa,&bb,&cc);
        add_edge(aa,bb,cc);
        add_edge(bb,aa,cc);
        add_edge2(cc,aa,bb);
    }
    dfs(s);
    fo(i,1,n){
        printf("%d:",i);
        if (p[i])
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}