洛谷——P1475 控制公司 Controlling Companies

P1475 控制公司 Controlling Companies

题目描述

有些公司是其他公司的部分拥有者，因为他们获得了其他公司发行的股票的一部分。(此处略去一句废话）据说，如果至少满足了以下三个条件之一，公司A就可以控制公司B了：

公司A = 公司B。

公司A拥有大于50%的公司B的股票。

公司A控制K(K >= 1)个公司，记为C1, ..., CK，每个公司Ci拥有xi%的公司B的股票，并且x1+ .... + xK > 50%。

给你一个表，每行包括三个数(i,j,p)；表明公司i享有公司j的p%的股票。计算所有的数对(h,s)，表明公司h控制公司s。至多有100个公司。

写一个程序读入N组数(i,j,p)，i,j和p是都在范围(1..100)的正整数，并且找出所有的数对(h,s)，使得公司h控制公司s。

一个递归的过程，DFS

#include<bits/stdc++.h>

using namespace std;

int n,v[505][505],m,cnt[505];

bool own[505],vi[505];

void dfs(int x){
    if(vi[x]) return;
    vi[x]=true;
    for(int i=1;i<=m;i++){
        cnt[i]+=v[x][i];
        if(cnt[i]>50){
            own[i]=true;
            dfs(i);
        }
    }
}

int main()
{
    scanf("%d",&n);
    for(int x,y,p,i=1; i<=n; i++) {
        scanf("%d%d%d",&x,&y,&p);
        v[x][y]=p;
        m=max(x,max(m,y));
    }
    
    for(int i=1;i<=m;i++){
        memset(own,0,sizeof(own));
        memset(vi,0,sizeof(vi));
        memset(cnt,0,sizeof(cnt));
        dfs(i);
        for(int j=1;j<=m;j++)
            if(i!=j&&own[j]) printf("%d %d
",i,j);
    }
    return 0;
}