Gym 101128A Promotions （拓扑序性质+DFS标记）

题目链接：http://codeforces.com/gym/101128/attachments

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll unsigned long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
#define mst(a,b) memset((a),(b),sizeof(a))
#define pii pair<int,int>
#define fi first
#define se second
#define mk(x,y) make_pair(x,y)
const int mod=1e9+7;
const int maxn=5e3+100;
const int maxm=1e5+100;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意：给定一个有向无环图，
现在要选择a~b人，
问选择a个人其有哪几个人是必选的，
选择b个人其有哪几个人是必选的，
有多少人是一定不会被选的。

题目分析：分析题目中的逻辑关系和拓扑序的性质，
如果一个人一定会被选到，，假设这个人影响到x个人，
那么n-x一定小于等于计划选拔人数，因为这个人如果不被选
那么被其影响到的人一定不会被选，而除此之外的人，
因为拓扑序的性质，肯定可以枚举完。
按照这个思路再分析一定不会被选到的人，如果影响到其的总人数
大于等于目标人数那么这个人肯定不会被选到。
在搜索中有标记的技巧，其每次搜索标记和当前搜索顺序相关。
复杂度：O(n).
*/

int a,b,e,p,x,y;
vector<int> g[maxn],v[maxn];
int d[maxn],vis[maxn],num[maxn];
int dfs1(int u,int v){
    if(vis[u]==v+1) return 0;
    vis[u]=v+1;
    int ret=1;
    rep(i,0,g[u].size()) ret+=dfs1(g[u][i],v);
    return ret;
}
int dfs2(int u,int p){
    if(vis[u]==p+1) return 0;
    vis[u]=p+1;
    int ret=1;
    rep(i,0,v[u].size()) ret+=dfs2(v[u][i],p);
    return ret;
}
int main(){
    scanf("%d%d%d%d",&a,&b,&e,&p);
    rep(i,0,p){
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        v[y].push_back(x);
        d[y]++;
    }
    rep(i,0,e) num[i]=dfs1(i,i);
    int ans1=0,ans2=0,ans3=0;
    rep(i,0,e){
        if(e-num[i]<a) ans1++;
        if(e-num[i]<b) ans2++;
    }
    mst(vis,0);
    rep(i,0,e){
        int tmp=dfs2(i,i);
        if(tmp>b) ans3++;
    }
    printf("%d %d %d\n",ans1,ans2,ans3);
    return 0;
}