题目链接:https://vjudge.net/problem/Gym-101128A
一看到这个题就想到了拓扑排序,暴搜出所有的拓扑序列,求出所有拓扑序列前A个数的交集,前B个数的交集,后n-B个数的交集,我无脑暴搜了一发,T,然后加了个set优化找拓扑序列的过程,又T,才明白所有的拓扑序列太多了,暴搜肯定T,又想我们只需要找出每个数在所有拓扑序列中最早出现的位置和最晚出现的位置,最早出现的位置比B大,肯定不会出现在前B个里面,最晚出现的位置小于等于A,肯定会出现前A个里面,找最早出现的位置需要建立反向图搜索,最晚出现的位置直接在正向图上搜索,搜索的时候要标记,避免重复计数。
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 5005;
vector<int> G[MAXN],_G[MAXN];
bool vis[MAXN];
int DFS1(int u)
{
if(vis[u]) return 0;
vis[u] = true;
int res = 1;
for(int i = 0; i < G[u].size(); i++) {
res += DFS1(G[u][i]);
}
return res;
}
int DFS2(int u)
{
if(vis[u]) return 0;
vis[u] = true;
int res = 1;
for(int i = 0; i < _G[u].size(); i++) {
res += DFS2(_G[u][i]);
}
return res;
}
int main(void)
{
int A,B,n,m;
int u,v;
int num,ans1,ans2,ans3;
while(scanf("%d %d %d %d",&A,&B,&n,&m) != EOF) {
for(int i = 0; i < n; i++) {
G[i].clear();
_G[i].clear();
}
ans1 = ans2 = ans3 = 0;
for(int i = 1; i <= m; i++) {
scanf("%d %d",&u,&v);
G[u].push_back(v);
_G[v].push_back(u);
}
for(int i = 0; i < n; i++) {
memset(vis,false,sizeof(vis));
num = n - DFS1(i);
if(num < A) ans1++;
if(num < B) ans2++;
}
for(int i = 0; i < n; i++) {
memset(vis,false,sizeof(vis));
num = DFS2(i);
if(num > B) ans3++;
}
printf("%d\n%d\n%d\n",ans1,ans2,ans3);
}
return 0;
}