POJ1236 Network of schools

#include<cstdio>
#include<algorithm>
#include<vector>
#include<stack>
using namespace std;
const int maxn=1014;
vector<int> g[maxn];
int N,M,x,y;
int low[maxn];
int dfn[maxn];
int cnt;
int scc;
int pos[maxn];
int in[maxn];
int out[maxn];
stack<int> st;
void tarjan (int x) {
    low[x]=dfn[x]=++cnt;
    st.push(x);
    for (int i=0;i<g[x].size();i++) {
        if (!low[g[x][i]]) {
            tarjan(g[x][i]);
            low[x]=min(low[x],low[g[x][i]]);
        }
        else if (!pos[g[x][i]])low[x]=min(low[x],dfn[g[x][i]]);
    }
    if (low[x]==dfn[x]) {
        scc++;
        while (1) {
            int u=st.top();
            st.pop();
            low[u]=low[x];
            pos[u]=scc;
            if (u==x) break;
        }
    }
} 
void build () {
    for (int i=1;i<=N;i++) {
        for (int j=0;j<g[i].size();j++) {
            if (pos[i]!=pos[g[i][j]]) {
                out[pos[i]]++;
                in[pos[g[i][j]]]++;
            }
        }
    }
}
int main () {
    scanf ("%d",&N);
    for (int i=1;i<=N;i++) {
        while (scanf("%d",&x)&&x) g[i].push_back(x);
    }
    for (int i=1;i<=N;i++) 
    if (!low[i]) tarjan(i);
    build ();
    int rd=0;
    int cd=0;
    for (int i=1;i<=scc;i++) {
        if (in[i]==0) rd++;
        if (out[i]==0) cd++;
    }
    if (scc==1) printf ("1\n0\n");
    else printf ("%d\n%d\n",rd,max(rd,cd));
    return 0;
}