Problem A: Promotions

• 题意 

题目给定一个有向无环图，对于每一个u->v  只有在u升职的情况下v才能升职，给定A,B  分别询问在A次升职机会以及B次升职机会的情况下，一定能升职的人数，以及B次升职机会下一定不能升职的人数

• 思路

对于每一个点，最少需要几次机会才能升职呢？显然必须是所有能走到这个点的点都要升职完，这个点才能升职，统计所有能走到这个点的点的个数    O(n）

对于每一个点,最多需要几次机会才能升职呢?   显然是除了这个点能走到的点之外的点都升职位完成了，再升职这个点，统计这个点能走到的点的个数 O(n)

sum[n]:有多少个点能到这个点

sum_to[n]:这个点能到多少个点

在A次机会的情况下,如果(E-sum_to[n])<=A,那么这个点必然能升职位

B同理

如果sum[i]>B那么这个点必然不能升职

代码如下:

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define x first
#define y second
#define int long long 
#define debug cout << endl << "------------------------" << endl
typedef long long LL;
typedef pair<int,int>PII;
typedef pair<PII,int>PIII;
const int N=2e5+5;
const int M=1e6+5;
const int mod=998244353;
int n,m,k,q;
int d[N];
int h[N],e[M],ne[M],idx;
int A,B,E,P;
int sum[N];//多少个点能到达这个点
int sum_to[N];//这个点能到达多少个点
bool stt[N];
int dd[N];
vector<int>ans1;
vector<int>ans2;
vector<int>ans3;
void add(int a,int b){
	e[idx]=b;
	ne[idx]=h[a];
	h[a]=idx++;
}
void dfs(int x,int fa){
	for(int i=h[x];~i;i=ne[i]){
		int j=e[i];
		if(!stt[j])
		{	
		    stt[j]=true;
			sum[j]++;
			sum_to[fa]++;
			dfs(j,fa);
		}
	}
}
void solve() {
	memset(h,-1,sizeof h);
	cin>>A>>B>>E>>P;
	while(P--){
		int a,b;
		cin>>a>>b;
		dd[b]++;
		add(a,b);
	}
	for(int i=0;i<E;i++){
		memset(stt,false,sizeof stt);
		dfs(i,i);
		sum[i]++;
	}
	for(int i=0;i<E;i++){
		sum_to[i]=E-sum_to[i];
		if(sum_to[i]<=A)ans1.pb(i);
		if(sum_to[i]<=B)ans2.pb(i);
		if(sum[i]>B)ans3.pb(i);
	}
	cout<<ans1.size()<<endl<<ans2.size()<<endl<<ans3.size();
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int t=1;
	//cin >> t ;
	
	while(t--) {
		solve();
	}

	return 0;
}