Problem A: Promotions

最新推荐文章于 2024-09-14 22:25:23 发布

Wzp*

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量60

点赞数 1

分类专栏：题解文章标签：算法 dfs

本文链接：https://blog.csdn.net/qq_41330204/article/details/130344911

版权

题解专栏收录该内容

1 篇文章 0 订阅

订阅专栏

该文描述了一个基于有向无环图(DAG)的问题，其中每个节点的升职依赖于其上游节点。给定每个节点升职的次数限制，文章通过深度优先搜索计算在特定机会下能确保升职和无法确保升职的节点数量。算法主要关注每个节点的可达节点数和能影响的节点数。

摘要由CSDN通过智能技术生成

题意

题目给定一个有向无环图，对于每一个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;
}