标签：cnt int 题解 head P2999 ++ MAXN 节点

题目链接

思路：拓扑排序

对于牛奶来说，最多只有一种方式从一个接口流到另一个接口。

则不会有牛奶分开又聚到一起，故有一个性质：除非节点的入度 \(=0\)，否则任何出度 \(>1\) 的节点的子节点都不能放置混合器。

然后跑拓扑排序即可。

代码

#include <iostream>
#include <cstdio>
#include <queue>
#define int long long
using namespace std;

const int MAXN = 1e5 + 5;

int n, cnt;
int head[MAXN], in[MAXN], out[MAXN], milk[MAXN];
bool ismilk[MAXN];
queue<int> q;

struct edge
{
	int to, nxt;
}e[MAXN];

inline void add(int u, int v)
{
	e[++cnt] = edge{v, head[u]};
	head[u] = cnt;
}

inline void topo()
{
	cnt = 0; 
	for (int i = 1; i <= n; i++)
	{
		if (!in[i])
		{
			q.push(i);
			ismilk[i] = true; //挤奶器
			milk[i] = 1; //有多少个挤奶器的牛奶流过这里
			cnt++; //统计挤奶器的数量
		}
	}
	while (!q.empty())
	{
		int u = q.front();
		q.pop();
		if (out[u] > 1) //出度>1不用管
		{
			continue;
		}
		for (int i = head[u]; i; i = e[i].nxt)
		{
			int v = e[i].to;
			in[v]--;
			milk[v] += milk[u]; //加上父节点的值
			if (!in[v]) //父节点全部处理完
			{
				q.push(v);
			}
		}
	}
}

signed main()
{
	scanf("%lld", &n);
	for (int i = 1; i < n; i++)
	{
		int u, v;
		scanf("%lld%lld", &u, &v);
		add(u, v);
		out[u]++;
		in[v]++;
	}
	topo();
	for (int i = 1; i <= n; i++)
	{
		if (!ismilk[i] && milk[i] == cnt) //不是挤奶器并且所有挤奶器的牛奶流过这里
		{
			printf("%lld\n", i);
		}
	}
	return 0;
}

标签：cnt,int,题解,head,P2999,++,MAXN,节点
来源： https://www.cnblogs.com/mango13/p/solution-p2999.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。