洛谷：P2746 [USACO5.3]校园网Network of Schools（Tarjan联通分量+缩点）

题目描述

一些学校连入一个电脑网络。那些学校已订立了协议：每个学校都会给其它的一些学校分发软件（称作“接受学校”）。注意即使 B 在 A 学校的分发列表中，A 也不一定在 B 学校的列表中。

你要写一个程序计算，根据协议，为了让网络中所有的学校都用上新软件，必须接受新软件副本的最少学校数目（子任务 A）。更进一步，我们想要确定通过给任意一个学校发送新软件，这个软件就会分发到网络中的所有学校。为了完成这个任务，我们可能必须扩展接收学校列表，使其加入新成员。计算最少需要增加几个扩展，使得不论我们给哪个学校发送新软件，它都会到达其余所有的学校（子任务 B）。一个扩展就是在一个学校的接收学校列表中引入一个新成员。

输入格式

输入文件的第一行包括一个正整数 N，表示网络中的学校数目。学校用前 N 个正整数标识。
接下来 N 行中每行都表示一个接收学校列表（分发列表），第 i+1 行包括学校 i 的接收学校的标识符。每个列表用 0 结束，空列表只用一个 0 表示。

输出格式

你的程序应该在输出文件中输出两行。

第一行应该包括一个正整数，表示子任务 A 的解。

第二行应该包括一个非负整数，表示子任务 B 的解。

输入

5
2 4 3 0
4 5 0
0
0
1 0

输出

1
2

#include<bits/stdc++.h>
using namespace std;
#define pi acos(-1)
#define mod 80112002
#define ll long long
#define ull unsigned long long
#define mem(a) memset(a,0,sizeof(a))
#define cio ios::sync_with_stdio(false);
vector<int>e[110];
int dfn[110];
int low[110];
int ru[110];
int chu[110];  
int vis[110];
int color[110]; // 标记每个点所属的强联通分量
int step = 0;
int cnt = 0;
stack<int>s;
void tarjan(int x)  // Tarjan强联通分量
{   
    low[x] = dfn[x] = ++step;
    s.push(x); // 入栈
    vis[x] = 1; // 标记入栈
    for(int i = 0; i < e[x].size(); i++){ // 遍历相邻节点
        int p = e[x][i]; 
        if(!dfn[p]){
            tarjan(p);
            low[x] = min(low[x], low[p]);
        }else if(vis[p]){
            low[x] = min(low[x], dfn[p]);
        }
    }
    if(low[x]==dfn[x]){ //x是强连通分量的根
        cnt++; 
        int v;
        do{
            v = s.top(); // 取栈顶元素
            color[v] = cnt;
            vis[v] = 0; // 标记出栈
            s.pop(); // 出栈
        }while(v!=x);
    }
}
int a[10010][3]; // 记录每条边，便于统计入度出度
int main()
{   
    int n;
    cin >> n;
    int k = 0;
    for(int i = 1; i <= n; i++){
        int x;
        while(cin>>x){
            if(x==0) break;
            k++;
            e[i].push_back(x);
            a[k][1] = i;
            a[k][2] = x;
        }
    }
    for(int i = 1; i <= n; i++){
        if(!dfn[i]) tarjan(i);
    }
    for(int i = 1; i <= k; i++){   // 缩点
        if(color[a[i][1]]!=color[a[i][2]]){ // 
            chu[color[a[i][1]]]++;  
            ru[color[a[i][2]]]++;   
        }
    }
    int c1 = 0;
    int c2 = 0;
    for(int i = 1; i <= cnt; i++){
        if(ru[i]==0) c1++;
        if(chu[i]==0) c2++;
    }
    if(cnt==1){ // 只存在一个强连通分量，只需要一次，并且不需要加边
        cout << 1 << endl << 0 << endl;
    }else{
        cout << c1 << endl << max(c1,c2) << endl;
    }
    return 0;
}