CSU1081有向图BFS

集训队分组

Description
中南大学ACM的暑期集训马上就要开始了，这次集训会将全体N名集训队员（编号分别为1, 2, …, N）按集训选拔赛的排名分成两组，前K名队员分入A组，其余队员分入B组。

但现在助理教练CSGrandeur一不小心把集训选拔赛的排名弄丢了，而之前又没将A组和B组的人员确定出来，于是CSGrandeur打算问一下集训人员他们的名次各是怎样的，以此来确定一下A组的队员。

然而集训队员们都视名次如粪土，只是隐约记得某些人排在了自己的后面，最终反馈到CSGrandeur这里的一共有M条信息，每条信息都可以用一个二元组(x, y) (x!=y)表示，含义为第x名队员记得第y名队员的排名比自己的要靠后。

现在CSGrandeur想知道，根据这M条信息，是否可以确定出A组的队员呢？（默认所有集训队员反映的信息都是符合事实的。）

Input
输入包含多组测试数据。

对于每组测试数据，第一行包含三个正整数N (2<=N<=1000)、K (1<=K<=N)、M (1<=M<=10000)，含义同上。接下来M行每行有两个正整数x、y (1<=x, y<=N且x!=y)，分别描述了M条信息，对于每对x、y，均表示第x名队员记得第y名队员的排名比自己的要靠后。

Output
对于每组测试数据，如果可以确定出A组的队员，输出“YES”（不包括引号），否则输出“NO”（不包括引号）。

Sample Input

3 1 2
1 2
1 3

3 2 2
1 2
1 3

Sample output

YES
NO

Hint

思路：做一个邻接表搜索所有的能确定自己能在多少人前方，如果后点大于等于n-m，那么就认为在前m名。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int first[maxn],next[maxn],edge[maxn][2];
int vis[maxn];
int add(int a,int b,int c)
{
    next[c]=first[a];//next用于指向a上一次出现时的信息；
    first[a]=c;//标记目前a出现在c条信息
    edge[c][0]=a;
    edge[c][1]=b;
}
queue<int> q;
int bfs(int i)
{

    memset(vis,0,sizeof(vis));
    while(!q.empty())
        q.pop();
    q.push(i);
    int cnt=0;
    vis[i]=1;
    while(!q.empty())
    {
        int t = q.front();
        q.pop();
        for(int j=first[t];j!=-1;j=next[j])
        {
            if(vis[edge[j][1]]==0)
            {
                vis[edge[j][1]]=1;
                q.push(edge[j][1]);
                cnt++;
            }
        }
    }
    return cnt;
}
int main()
{
    int n,m,k;
    int xx,yy;
    while(scanf("%d %d %d",&n,&m,&k)!=EOF)
    {
        memset(first,-1,sizeof(first));
        memset(next,0,sizeof(next));
        int cnt1=0;
        for(int i=0;i<k;i++)
        {
            scanf("%d %d",&xx,&yy);
            add(xx,yy,i);
        }
        for(int i=1;i<=n;i++)
        {
            if(bfs(i)>=(n-m))
            {
                cnt1++;
            }
        }
        if(cnt1>=m)
            printf("YES\n");
        else
            printf("NO\n");
    }
}