luogu P4017 最大食物链计数(拓扑排序变形)

在这里插入图片描述
在这里插入图片描述
一个讲的很清楚的视频:链接

总之这是一个基于有向无环图的一个算法,目的是得到一个拓扑序列,拓扑序列就是每个指向某元素的元素一定在那个元素之前的一种有序序列,主要利用stack或者queue实现,先寻找入度为0的元素,将这些元素入队/入栈,然后当队列/栈里面有元素的时候,就每次取最头上的元素,然后让它指向的元素的入度都减1,之后将这个元素记录下来,如果在这个过程中又出现了入度为0的元素,就让他入队/入栈,一致重复这个过程,直到队列/栈为空为止,记录下来的有序元素序列就是一种拓扑序列(拓扑序列不一定唯一)。

下面是代码:

#include<iostream>
#include<stack>
using namespace std;
const int mod = 80112002;
int ans[5005];//存放到达某点时的总步数 
int enter[5005],leave[5005];//记录一个点的入度和出度 
bool map[5005][5005];//图的邻接矩阵储存法 
int sum=0;//最后的答案 
stack<int> st;
int main()
{
	int n,m;
	int a,b;
	int mi;
	cin>>n>>m;
	for(int i=1;i<=m;++i)
	{
		cin>>a>>b;
		map[a][b]=1;
		leave[a]++;//记录入度和出度 
		enter[b]++;
	}
	for(int i=1;i<=n;++i)
	{
		if(!enter[i])//若入度为0则让其入栈,并更新ans数组 
		{
			ans[i]=1;
			st.push(i);
		}
	}
	while(!st.empty())
	{
		mi=st.top();
		st.pop();
		for(int i=1;i<=n;++i)
		{
			if(map[mi][i])
			{
				ans[i]+=ans[mi];//更新ans数组 
				ans[i]%=mod;//记得取模 
				map[mi][i]=0;//剪掉出边,这里好像没必要 
				leave[mi]--;//更新两点的出入度 
				enter[i]--;
				if(enter[i]==0)
				{
					if(leave[i]==0)//如果找到了最高级消费者 
					{
						sum+=ans[i];//更新sum 
						sum%=mod;//记得取模 
						continue;//下一个 
					}
					st.push(i);//如果还不是最高级消费者,就入栈 
				}
			}
		}
	}
	cout<<sum<<endl;//输出答案 
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值