ccf 高速公路(顺便当tarjan模板）

tarjan算法的模板题，一开始80分，是因为我忘了从每个点都要开始遍历防止多个连通子图的出现

#include<bits/stdc++.h>
using namespace std;
vector<int> edge[10010];
stack<int>s;
bool vis[10010];
int num,sum;
int dfn[10010];
int low[10010];
void tarjan(int u)
{
int v;
dfn[u]=low[u]=num++;
vis[u]=true,s.push(u);
for(int e=0;e<edge[u].size();e++)
{
v=edge[u][e];
if(!dfn[v])
{
tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(vis[v])
low[u]=min(low[u],dfn[v]);
}
if(dfn[u]==low[u])
{
int cnt=0;
do
{
v=s.top();
vis[v]=false;
s.pop();
cnt++;
}while(v!=u);
sum+=cnt*(cnt-1)/2;
}

}
int main()
{
int n,m;
int u,v;
scanf("%d %d",&n,&m);
for(int i=0;i<m;i++)
{
scanf("%d %d",&u,&v);
edge[u].push_back(v);
}
num=1;
for(int i=1;i<=n;i++)
tarjan(i);
printf("%d\n",sum);

}