POJ 1236 Network of Schools 强连通分量

题意：给一个有向图，需要实现两个任务：

• A：分配给一些点电脑，使得这些点能够运输到其他所有点，求出这些点个数的最小值
• B：为了使分配给任意节点电脑，都能运输到其他所有点，求出需要加的最少边数

思路：任务A相当于求出入度为0的点的个数，任务B则需要思考，由于连通分量中每个点必须至少有一条入边一条出边，故为了使得所有点连通起来，需要将出度为0的点连一条边到入度为0的点。而需要连的边的最小值，只需要求出入度为0的点的数量和出度为0的点的数量中的最大值即可。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<vector>
#include<queue>
#include<map>
#include<algorithm>
#include<stack>
using namespace std;
typedef long long ll;
const int maxn = 1e3+5;
const int inf = 0x3f3f3f3f;
int dfn[maxn], low[maxn], vis[maxn], cnt, cnt2, n, m, u[maxn], out[maxn], in[maxn];
vector<int> g[maxn];
stack<int> s;
void tarjan(int x)
{
	dfn[x] = low[x] = ++cnt;
	s.push(x);
	vis[x] = 1;
	for (int i = 0; i < g[x].size(); i++) {
		int v = g[x][i];
		if (!dfn[v]) {
			tarjan(v);
			low[x] = min(low[x], low[v]);
		}
		else if (vis[v]) 
			low[x] = min(low[x], dfn[v]);
	}
	if (low[x] == dfn[x]) {
		cnt2++;
		while (1) {
			int y = s.top(); s.pop();
			vis[y] = 0;
			u[y] = cnt2;
			if (y == x)
				break;
		}
	}
}
int main()
{
//	freopen("test.txt", "r", stdin);
	while (cin >> n) {
		memset(out, 0, sizeof(out));
		memset(dfn, 0, sizeof(dfn));
		memset(vis, 0, sizeof(vis));
		for (int i = 1; i <= n; i++) {
			int v;
			while (cin >> v && v) {
				g[i].push_back(v);
			}
		}
		cnt = cnt2 = 0;
		for (int i = 1; i <= n; i++)
			if (!dfn[i])
				tarjan(i);
		int ans1 = 0, ans2 = 0;
		for (int i = 1; i <= n; i++)
			for (int j = 0; j < g[i].size(); j++) {
				int v = g[i][j];
				if (u[i] != u[v]) {
					in[u[v]]++;
					out[u[i]]++;
				}
			}
		for (int i = 1; i <= cnt2; i++) {
			if (!in[i])
				ans1++;
			if (!out[i])
				ans2++;
		}
		ans2 = cnt2 == 1 ? 0 : max(ans2, ans1);
		printf("%d\n%d\n", ans1, ans2);
	}
	return 0;
}