CSUOJ 1081 集训队分组

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

建有向图，对每个结点bfs一遍，找出从它出发可以到达多少个结点，如果有n-k个结点在它之后，那它就是前k个了；

邻接矩阵会超时，要使用邻接表；

#include<stdio.h>
#include<string.h>
#include<vector>
#define N 1005
using namespace std;

vector<int> g[N];
char vis[N];

int bfs(int u)
{
    int cx, nx, ret = 0;
    int Q[N], front, rear;

    memset(vis, 0, sizeof(vis));
    Q[1] = u;
    vis[u] = 1;
    front = 1;
    rear  = 2;
    while(front < rear){
        cx = Q[front++];
        for(int i = 0; i < g[cx].size(); i++){
            nx = g[cx][i];
            if (!vis[nx]){
                Q[rear++] = nx;
                vis[nx] = 1;
                ret++;
            }
        }
    }
    return ret;
}

int main()
{
    int n, k, m;
    while(scanf("%d %d %d", &n, &k, &m) != EOF){
        int u, v;
        for(int i = 1; i <= n; i++)
            g[i].clear();
        for (int i = 0; i < m; i++){
            scanf("%d%d", &u, &v);
            g[u].push_back(v);
        }
        int cnt = 0;
        for(int i = 1; i <= n; i++){
            if(bfs(i) >= n-k)
                cnt++;
            if(cnt >= k)
                break;
        }
        if(cnt >= k)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}

PS：参考的大牛的博客，没看太懂……