强联通块tarjan算法

http://poj.org/problem?id=1236
第一问：需要几个学校存在软件，才能通过传递，使得所有的学校都有软件
用tarjan算法求出强联通分量后，将每个联通分量缩成一个点，那么问题1的答案就是入度为0的点的个数
为什么？入度为0的点，肯定不能通过其他学校传送软件给他，所以他必须存在一份软件
第二问：需要加几条边，才能使得图强联通
缩点后，a为所有入度为0的点的个数，b为所有出度为0的点的个数，那么答案就是max(a,b)

 #include <stdio.h>
 #include <string.h>
 #include <vector>
 #include <stack>
 using namespace std;
 const int N =  + ;
 vector<int> G[N];
 stack<int> st;
 bool vis[N];
 int sccno[N],pre[N],lowlink[N],in[N],out[N],dfs_clock,scc_cnt;
 int min(const int &a, const int &b)
 {
     return a < b ? a : b;
 }
 void tarjan(int u)
 {
     pre[u] = lowlink[u] = ++dfs_clock;
     st.push(u);
     for(int i=; i<G[u].size(); ++i)
     {
         int v = G[u][i];
         if(!pre[v])
         {
             tarjan(v);
             lowlink[u] = min(lowlink[u],lowlink[v]);
         }
         else if(!sccno[v])
             lowlink[u] = min(lowlink[u],pre[v]);
     }
     if(lowlink[u]==pre[u])
     {
         scc_cnt++;
         for(;;)
         {
             int x = st.top();st.pop();
             sccno[x] = scc_cnt;
             if(x==u) break;
         }
     }
 }
 void find_scc(int n)
 {
     for(int i=; i<=n; ++i)
         if(!pre[i])
             tarjan(i);
 }
 int main()
 {
     int n,i,a,b;
     scanf("%d",&n);
     for(a=; a<=n; ++a)
     {
         while(scanf("%d",&b),b)
         {
             G[a].push_back(b);
             vis[b] = true;
         }
     }
     find_scc(n);
     for(i=; i<=scc_cnt; ++i) in[i] = out[i] = ;
     for(int u=; u<=n; ++u)
         for(i=; i<G[u].size();++i)
         {
             int v = G[u][i];
             if(sccno[u] != sccno[v]) in[sccno[v]] = out[sccno[u]] = ;
         }
     a = b = ;
     for(i=; i<=scc_cnt; ++i)
     {
         if(in[i]) a++;
         if(out[i]) b++;
     }
     int ans = a > b ? a : b;
     if(scc_cnt == ) ans = ;
     printf("%d\n%d\n",a,ans);
     return ;
 }