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);
	
}