GYM 101128 A.Promotions（dfs）

Description
给出一张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
Solution
前两个值是同种求法，假设要提拔x个人2，那么一个人是必须被提拔的意思是如果不提拔这个人，那么所有被其支配的人不会被提拔，而剩下的人数不够x，所以对原图从每个点开始dfs，所有被其支配的点打上时间戳便于统计（不然可能会重），计算出每个点的num值表示其支配的人数（包括自身），如果n-num[i] < x说明只选拔x个人时i必须要被提拔
对于第三个值，在提拔b个人的条件下，一个人一定不会被提拔的意思是如果提拔这个人那么需要提拔他的一系列上司，而这些人的数量已经超过b了，所以对反图进行上述同样的dfs操作，得到的每个点的num值表示要提拔这个人至少要提拔的人数，如果num[i] > b说明这个人一定不会被提拔
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#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);
    }
    return 0;
}