USACO 2.3.4 Controlling Companies

题解

数据弱暴力搜索搞定。也可以用并查集和图论算法搞定。

---

Code

// head files excluded
using namespace std;
int n,m,k;

int f[101][101];
bool vis[101];
bool ctr[101];
int ct[101];
void dfs(int x){
    if(vis[x]) return;
    vis[x] = true;

    for(int i=1;i<=m;i++){
        ct[i]+=f[x][i];
        if(ct[i]>50){
            ctr[i] = true;
            dfs(i);
        }
    }
}

int main(void){

    cin>>n;
    int i,j,p;
    m=0;
    for(int k=0;k<n;k++){
        cin>>i>>j>>p;
        f[i][j] = p;
        m = max(max(i,j),m);
    }

    for(int i=1;i<=m;i++){
        memset(vis,false,sizeof(vis));
        memset(ctr,false,sizeof(ctr));
        memset(ct,0,sizeof(ct));
        dfs(i);
        for(int j=1;j<=m;j++){
            if(ctr[j] && j!=i) cout<<i<<" "<<j<<endl;
        }
    }

    return 0;
}