mcpc2017 Faulty Robot （bfs+dfs）

题目描述

As part of a CS course, Alice just finished programming her robot to explore a graph having n nodes, labeled 1, 2, . . . , n, and m directed edges. Initially the robot starts at node 1.
While nodes may have several outgoing edges, Alice programmed the robot so that any node may have a forced move to a specific one of its neighbors. For example, it may be that node 5 has outgoing edges to neighbors 1, 4, and 6 but that Alice programs the robot so that if it leaves 5 it must go to neighbor 4.
If operating correctly, the robot will always follow forced moves away from a node, and if reaching a node that does not have a forced move, the robot stops. Unfortunately, the robot is a bit buggy, and it might violate those rules and move to a randomly chosen neighbor of a node (whether or not there had been a designated forced move from that node). However, such a bug will occur at most once (and might never happen). 
Alice is having trouble debugging the robot, and would like your help to determine what are the possible nodes where the robot could stop and not move again.
We consider two sample graphs, as given in Figures G.1 and G.2. In these figures, a red arrow indicate an edge corresponding to a forced move, while black arrows indicate edges to other neighbors. The circle around a node is red if it is a possible stopping node.

In the first example, the robot will cycle forever through nodes 1, 5, and 4 if it does not make a buggy move.
A bug could cause it to jump from 1 to 2, but that would be the only buggy move, and so it would never move on from there. It might also jump from 5 to 6 and then have a forced move to end at 7.
In the second example, there are no forced moves, so the robot would stay at 1 without any buggy moves. It might also make a buggy move from 1 to either 2 or 3, after which it would stop.

输入

输出

样例输入

7 9
1 2
2 3
-1 5
2 6
5 1
-4 1
5 6
-6 7
-5 4

样例输出

2

#include <iostream>
#include <cstdio>
#include <vector>
#include <string.h>
#include <queue>
using namespace std;
vector<int> kkm[1100];
vector<int> jjc[1100];
int vis[1100];
int vist[1100];
int ans[1100];
int bck[1100];/*bfs能到达的点*/
int b=0;
int res=0;
void dfs(int sx)
{
    if(vist[sx]||vis[sx]) return;
    if(kkm[sx].size()==0) {ans[sx]=1;return;}/*标记停止的点*/
    vist[sx]=1;
    for(int i=0;i<kkm[sx].size();i++)
    {dfs(kkm[sx][i]);vist[sx]=0;}
}
void bfs()
{
    queue<int> q;
    vis[1]=1;
    bck[b++]=1;
    q.push(1);
    while(!q.empty())
    {
        int t=q.front();
        q.pop();
        if(kkm[t].size()==0) {res++;vis[t]=1;continue;}/*机器人还可能停止在红线连接的地图上*/
        for(int i=0;i<kkm[t].size();i++)
        {
            if(vis[kkm[t][i]]==1) continue;
            vis[kkm[t][i]]=1;
            q.push(kkm[t][i]);
            bck[b++]=kkm[t][i];
        }
    }
}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    int x,y;
    for(int i=0;i<=n;i++) {ans[i]=0;vis[i]=0;vist[i]=0;}
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&x,&y);
        if(x<0)    {kkm[-x].push_back(y);}/*红线地图*/
        else       {jjc[x].push_back(y);}/*黑线地图*/
    }
    bfs();
    for(int i=0;i<b;i++)
    {
        for(int j=0;j<jjc[bck[i]].size();j++)
        {
            dfs(jjc[bck[i]][j]);/*首次dfs按照黑线地图*/
        }
    }
    for(int i=1;i<=n;i++)
    {if(ans[i]==1) res++;}
    printf("%d\n",res);
    return 0;
}