CSU1081-集训队分组-BFS

C: 集训队分组

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

思路：将输入视为一个有向图，读入图对于每个人进行一次BFS看看它的后继节点有多少个，如果后继节点数大于等于n-k，则其为前k名，就看能不能有k个点

#include <bits/stdc++.h>
#define N 10100
#define INF 0x3f3f3f3f
#define LL long long
#define mem(a,n) memset(a,n,sizeof(a))
#define fread freopen("in.txt","r",stdin)
#define fwrite freopen("out.txt","w",stdout)
using namespace std;
vector<int> G[1010];
bool vis[1012];
void init(int n)
{
    for(int i=1;i<=n;++i){
        G[i].clear();
    }
}
int bfs(int beg)
{
    memset(vis,0,sizeof(vis));
    queue<int> q;
    q.push(beg);
    vis[beg]=true;
    int t=0;
    while(!q.empty()){
        int cur=q.front();
        q.pop();
        for(int i=0;i<G[cur].size();++i){
            if(!vis[G[cur][i]]){
                vis[G[cur][i]]=true;
                q.push(G[cur][i]);
                ++t;
            }
        }
    }
    return t;
}

int main()
{
    ios::sync_with_stdio(false);
    int n,k,d,c1,c2,cnt;
    while(cin>>n>>k>>d){
        init(n);
        cnt=0;
        for(int i=0;i<d;++i){
            cin>>c1>>c2;
            G[c1].push_back(c2);
        }
        for(int i=1;i<=n;++i){
            if(bfs(i)>=n-k){
                ++cnt;
            }
        }
        if(cnt>=k){
            cout<<"YES"<<endl;
        }else{
            cout<<"NO"<<endl;
        }
    }
    return 0;
}