集训队分组
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+;
int first[maxn],next[maxn],edge[maxn][];
int vis[maxn];
int add(int a,int b,int c)
{
next[c]=first[a];//next用于指向a上一次出现时的信息;
first[a]=c;//标记目前a出现在c条信息
edge[c][]=a;
edge[c][]=b;
}
queue<int> q;
int bfs(int i)
{ memset(vis,,sizeof(vis));
while(!q.empty())
q.pop();
q.push(i);
int cnt=;
vis[i]=;
while(!q.empty())
{
int t = q.front();
q.pop();
for(int j=first[t];j!=-;j=next[j])
{
if(vis[edge[j][]]==)
{
vis[edge[j][]]=;
q.push(edge[j][]);
cnt++;
}
}
}
return cnt;
}
int main()
{
int n,m,k;
int xx,yy;
while(scanf("%d %d %d",&n,&m,&k)!=EOF)
{
memset(first,-,sizeof(first));
memset(next,,sizeof(next));
int cnt1=;
for(int i=;i<k;i++)
{
scanf("%d %d",&xx,&yy);
add(xx,yy,i);
}
for(int i=;i<=n;i++)
{
if(bfs(i)>=(n-m))
{
cnt1++;
}
}
if(cnt1>=m)
printf("YES\n");
else
printf("NO\n");
}
}