Chocolate Milk S

题目

题目大意

让我们求一个所有入度为0的点都等达到的点

---

对于50%的数据

我们可以想到O(n*n)的解决思路。

在输入的时候，先预处理出每个点的入度和出度，再一重判断枚举每个点,如果是挤奶器，那么就遍历这个点，再是一重循环枚举管道和奶罐，如果这个点可以被所有的挤奶器流到，那么继续遍历这个点，看看这个点能不能到达所有的奶罐如果可以那么就可以在这个地方设立巧克力混合器，注意也可以在奶罐处设立巧克力混合器

#include<bits/stdc++.h>
#define maxn 100100
using namespace std;
inline int read()
{
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
	while(isdigit(ch)){res=(res<<1)+(res<<3)+(ch&15);ch=getchar();}
	return res*f;
}
int n,head[maxn],tot;
struct ad{
	int ne;
	int to;
}edge[maxn];
inline void add(int x,int y)
{
	edge[++tot].to=y;
	edge[tot].ne=head[x];
	head[x]=tot;
}
int rdu[maxn],cdu[maxn],li[maxn],ans[maxn],num;
int vis[maxn],p[maxn],q[maxn];
inline void dfs(int x)//找挤奶器可以通向的地方
{
	vis[x]=1;
	for(int i=head[x];i;i=edge[i].ne)
	if(!vis[edge[i].to])q[edge[i].to]++,dfs(edge[i].to);
}
int sum;
inline void dfs2(int x)//遍历是不是可以流到奶罐
{

	vis[x]=1;if(p[x]==2)sum++;
	for(int i=head[x];i;i=edge[i].ne)
	if(!vis[edge[i].to])dfs2(edge[i].to);	
	
} 
int ni,g;
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		int x,y;x=read();y=read();
        cdu[x]++;rdu[y]++;
        add(x,y);
	}
	for(int i=1;i<=n;i++)
	if(rdu[i]==0)p[i]=1,ni++;
	else if(cdu[i]==0)p[i]=2,g++;
	else p[i]=3;
	for(int i=1;i<=n;i++){if(p[i]==1){memset(vis,0,sizeof(vis));dfs(i);}}
    for(int i=1;i<=n;i++)
	{
    	if(p[i]!=1&&q[i]==ni)
    	{
    		sum=0;
    		memset(vis,0,sizeof(vis));
    		
			dfs2(i);
    		
    		if(sum==g)printf("%d\n",i);
			
		}
	}
	return 0;
}

对于100%的数据

我们O(n)解决（拓扑排序）
在输入的时候，预处理出每个点的出度和入度，然后一重循环枚举出挤奶处，把他压入队列,对于每个点如果它的出度>1，那么他所连的边就不可能成为巧克力混合器，因为这是一棵树。如果就只有一的出度，那么所指向的点加上这个点的奶流量，将那个点的入度–，如果入度等于0了，那么就可以入队了，最后只需要判断一下能不能流满就好

#include<bits/stdc++.h>
#define maxn 1000100
using namespace std;
inline int read()
{
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
	while(isdigit(ch)){res=(res<<1)+(res<<3)+(ch&15);ch=getchar();}
	return res*f;
}
int rdu[maxn],cdu[maxn],n;
int to[maxn],milk[maxn];
int vis[maxn];
int q[maxn];
int l=1,r;
int milkn;
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		int x,y;x=read();y=read();
		cdu[x]++;
		rdu[y]++;
		to[x]=y;
	}
	for(int i=1;i<=n;i++)
	if(!rdu[i]){vis[i]=1;q[++r]=i;milk[i]=1;milkn++;}//把初始的挤奶点入队
	
	while(l<=r)
	{
		int u=q[l];
	    if(cdu[u]>1)
	    {
	    	milk[to[u]]=0;l++;
	    	continue;
		}
	    milk[to[u]]+=milk[u];//可以流到
	    l++;
	    rdu[to[u]]--;
	    if(!rdu[to[u]])q[++r]=to[u];//入队
	}
	for(int i=1;i<=n;i++)
	if(!vis[i]&&milk[i]==milkn)printf("%d\n",i);
	return 0;
}