poj1236 Network of Schools

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
#include<stack>
using namespace std;
stack<int>S;
struct node
{
int id;
node *next;
}head[110];
int n;
int ranse;
int index;
int color[110];//每个节点的颜色
bool visit[110];
bool instack[110];
int dfn[110];
int low[110];
int colorrudu[110];//每种颜色的入度
int colorchudu[110];//每种颜色的出度
void init()
{
int i;
for(i=1;i<=n;i++)
{
instack[i]=false;
visit[i]=false;
colorrudu[i]=0;
colorchudu[i]=0;
}
}
void dfs(int u)
{
S.push(u);
instack[u]=true;
visit[u]=true;
dfn[u]=low[u]=++index;
node *p;
p=head[u].next;
while(p!=NULL)
{
if(!visit[p->id])
{
dfs(p->id);
low[u]=low[u]<low[p->id]?low[u]:low[p->id];
}
else if(instack[p->id])
{
low[u]=low[u]<low[p->id]?low[u]:low[p->id];
}
p=p->next;
}
int v;
if(dfn[u]==low[u])
{
ranse++;
while(1)
{
v=S.top();S.pop();
color[v]=ranse;
instack[v]=false;
if(v==u)
{
break;
}
}
}
}
void caldu()
{
node *p;
int i;
int ourcolor;
for(i=1;i<=n;i++)
{
ourcolor=color[head[i].id];
p=head[i].next;
while(p!=NULL)
{
if(color[p->id]!=ourcolor)
{
colorchudu[ourcolor]++;
colorrudu[color[p->id]]++;
}
p=p->next;
}
}
}
int main()
{
scanf("%d",&n);
int i;
node *temp;
for(i=1;i<=n;i++)
{
head[i].id=i;
int b;
while(scanf("%d",&b),b)
{
temp=head[i].next;
head[i].next=new node;
head[i].next->id=b;
head[i].next->next=temp;
}
}
//输入完毕
init();
index=0;
ranse=0;
for(i=1;i<=n;i++)
{
if(!visit[i])
{
dfs(i);
}
}
//染色完毕，一种颜色为一个强连通分量
caldu();//计算每种 颜色的入度和出度
if(ranse==1)
{
printf("1\n0\n");
return 0;
}
int res1=0,res2=0;
for(i=1;i<=ranse;i++)
{
if(colorrudu[i]==0)
{
res1++;
}
if(colorchudu[i]==0)
{
res2++;
}
}
printf("%d\n",res1);
printf("%d\n",res1>res2?res1:res2);
return 0;
}