GYM 101128 A.Promotions（dfs）

给出一张n个点的有向图表示一家有n个员工的公司的隶属图，u->v表示u是v的上司，现在老板要提拔一些人，但是规定如果一个员工被提拔，那么他的上司也要被提拔，现给出两个整数a和b表示一区间，求三个值，第一个值表示如果提拔a个人那么这n个中必须要被提拔的人数，第二个值表示如果提拔b个人那么这n个人中必须要被提拔的人数，第三个值表示就算提拔b个人也不会被提拔的人数
Input
第一行四个整数n,m,a,b分别表示人数，关系数以及提拔的人数区间，之后m行每行两个整数u和v表示u是v的上司(2<=n<=5000,1<=m<=10000,1<=a < b < n)
Output
输出三个要求的值
Sample Input
3 4 7 8
0 4
1 2
1 5
5 2
6 4
0 1
2 3
4 5
Sample Output
2
4
3

因为下属个数是确定的。所以先正向建图求出下属 再反向建图求出上属，如果录取人数为a，那么n-num[i]

#include <bits/stdc++.h>
using namespace std;
#define maxn 5555
int a,b,n,m,num[maxn],vis[maxn];
vector<int> g[maxn],gg[maxn];
int dfs1(int u,int fa)
{
    if(vis[u]==fa+1) return 0;
    vis[u] = fa+1;
    int ans=1;
    for(int i=0;i<g[u].size();i++) ans+=dfs1(g[u][i],fa);
    return ans;
}

int dfs2(int u,int fa)
{
    if(vis[u]==fa+1) return 0;
    vis[u]=fa+1;
    int ans=1;
    for(int i=0;i<gg[u].size();i++)
        ans+=dfs2(gg[u][i],fa);
    return ans;
}
int main()
{
    while(~scanf("%d%d%d%d",&a,&b,&n,&m))
    {
        for(int i=0;i<n;i++) g[i].clear(),gg[i].clear();
        while(m--)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            g[u].push_back(v);
            gg[v].push_back(u);
        }
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++) num[i]=dfs1(i,i);
        int ans1=0,ans2=0,ans3=0;
        for(int i=0;i<n;i++)
        {
            if(n-num[i]<a) ans1++;
            if(n-num[i]<b) ans2++;

        }
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++) num[i]=dfs2(i,i);
        for(int i=0;i<n;i++)
        {
            if(num[i]>b) ans3++;
        }
        printf("%d\n%d\n%d\n",ans1,ans2,ans3 );
    }
}