USACO2.3.5 Controlling Companies (concom)

DFS

a[i,j] 记录i控制j的股份

对于每个公司i，搜索i直接控股的所有公司，将股份加到cnt[i]上，如果超过50，则：

1.own[j]=1，表示i控制了j

2.递归搜索公司j，每个公司至多只递归一次，所以用vis判重

每搜完一个i公司，扫一遍own[j]，输出控制了哪些公司

/*
ID:xsy97051
LANG:C++
TASK:concom
*/
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std; 

bool vis[101],own[101];
int a[101][101],cnt[101],m,n;

void dfs(int x)
{
	if(vis[x]) return;
	vis[x]=1;
	for(int i=m;i>0;--i)
	{
		cnt[i]+=a[x][i];
		if(cnt[i]>50)
		{
			own[i]=1;
			dfs(i);
		}
	}
}

int main()
{
	freopen("concom.in","r",stdin);
	freopen("concom.out","w",stdout);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		a[u][v]+=w;
		m=max(m,max(u,v));
	}
	
	for(int i=1;i<=m;++i)
	{
		memset(vis,0,sizeof(vis));
		memset(own,0,sizeof(own));
		memset(cnt,0,sizeof(cnt));
		dfs(i);
		for(int j=1;j<=m;j++)
		if(i!=j && own[j])
		cout<<i<<" "<<j<<endl;
	}
	return 0;
}