poj1236 Network of Schools（tarjan缩点）

强连通缩点重构图，入度为0的点的个数即为第一问的答案，第二问的答案为入度为0的点的个数和出度为0的点的个数的最大值。证明见百度。

#include <cstdio>#include <cstring>#include <algorithm>#include <stack>using namespace std;#define ll long long#define inf 0x3f3f3f3f#define N 110#define pa pair<int,int>inline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();    return x*f;}int n,dfn[N],low[N],dfnum=0,bel[N],scc=0,in[N],out[N];bool inq[N],mp[N][N],mp1[N][N];stack<int>qq;void tarjan(int x){    dfn[x]=low[x]=++dfnum;qq.push(x);inq[x]=1;    for(int y=1;y<=n;++y){        if(!mp[x][y]) continue;        if(!dfn[y]) tarjan(y),low[x]=min(low[x],low[y]);        else if(inq[y]) low[x]=min(low[x],dfn[y]);    }    if(dfn[x]==low[x]){        ++scc;while(1){            int y=qq.top();qq.pop();inq[y]=0;            bel[y]=scc;if(y==x) break;        }    }}void rebuild(){    for(int x=1;x<=n;++x)        for(int y=1;y<=n;++y){            if(!mp[x][y]) continue;            if(bel[x]!=bel[y]&&!mp1[bel[x]][bel[y]])                mp1[bel[x]][bel[y]]=1,in[bel[y]]++,out[bel[x]]++;        }}int main(){//  freopen("a.in","r",stdin);    n=read();    for(int i=1;i<=n;++i){        while(1){            int x=read();if(!x) break;mp[i][x]=1;        }    }    for(int i=1;i<=n;++i) if(!dfn[i]) tarjan(i);    rebuild();int ans1=0,ans2=0;//ans1---入度为0的个数，ans2---出度为0的个数     for(int i=1;i<=scc;++i){        if(in[i]==0) ans1++;        if(out[i]==0) ans2++;    }    if(scc==1){puts("1");puts("0");return 0;}    printf("%d\n%d\n",ans1,max(ans1,ans2));    return 0;}