CSU 1081集训队分组（搜索）

集训队分组

Time Limit: 2 Sec   Memory Limit: 128 MB

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

解法：从点x开始广搜，如果从x可以到达的点有v个，说明有v个点在x后面，如果v大于N-K，说明x在前k名中。

#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;

const int MaxN = 1005;
int vis[MaxN];
vector <int> Map[MaxN];

//广搜，返回从x出发可以到达多少个点，即有多少个点在它后面
int BFS(int x) {
    queue <int> q;
    q.push(x);
    memset(vis, 0, sizeof(vis));
    vis[x] = 1;
    int cnt = 0;
    while(!q.empty()) {
        int nx = q.front(); q.pop();
        for(int i = 0; i < Map[nx].size(); ++i) {
            int xx = Map[nx][i];
            if(!vis[xx]) {
                q.push(xx);
                vis[xx] = 1;
                ++cnt;
            }
        }
    }
    return cnt;
}

int main() {
    int N, K, M, x, y;
    while(~scanf("%d%d%d", &N, &K, &M)) {
        memset(Map, 0, sizeof(Map));
        for(int i = 0; i < M; ++i) {
            scanf("%d%d", &x, &y);
            Map[x].push_back(y);
        }
        int cnt = 0;
        for(int i = 1; i <= N; ++i) {
            if(BFS(i) >= N - K) ++cnt; //如果至少有N-K个结点在它之后，说明它就是前K个
            if(cnt == K) break;
        }
        if(cnt == K) puts("YES");
        else puts("NO");
    }
    return 0;
}