201509-4-高速公路

(http://115.28.138.223/view.page?gpid=T29)

//邻接表存图 
#include <stdio.h>
#include <string.h>
#include <iostream>
#define maxn 200010
using namespace std;

int dfn[maxn],low[maxn]; 
int clock,top; //当前元素是被访问的序号,模拟栈的首部
bool instack[maxn];
int stack[maxn],head[maxn];
int ans,tot;

struct Node 
{
    int to;//表头节点邻接顶点
    int next;//指向下一个邻接顶点 
}edge[maxn];
void addEdge(int u,int v)
{ 
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++; //head[u]里存储的是最后一条以u为顶点的边的序号
}
void tarjan(int u) 
{
    dfn[u]=low[u]=clock++; //初始化该元素的dfn和low数组
    instack[u]=true; //入栈
    stack[++top]=u;
    for(int i=head[u];i!=-1;i=edge[i].next)
    { //遍历所有以该点为顶点的边
        int v=edge[i].to;
        if(!dfn[v]) 
        { //如果这个点没有被访问过
            tarjan(v); //继续向下搜索
            low[u]=min(low[u],low[v]); //更新当前点能回溯到的最远点
        }
//如果已经被访问过 没有被删除 说明这个点是某个强连通分量的一点 当前"根"的low值和当前点的dfn值比较并更新
        else if(instack[v]&&low[u]>dfn[v])
            low[u]=dfn[v]; 
    }
    if(low[u]==dfn[u]) 
    { // 找到"根"取出当前强连通分量
        int temp=0;
        int k;
        do 
        {
            k=stack[top--];
            instack[k]=false;
            temp++;
        }while(k!=u);
        if (temp>1) 
            ans+=(temp*(temp-1))/2;
    }
}
int main() 
{
    int n,m;
    scanf("%d%d",&n,&m); 
    memset(head,-1,sizeof(head));
    clock=0,tot=0;
    top=-1,ans=0;
    for(int i=0;i<m;++i) 
    {
        int u,v;
        scanf("%d%d",&u,&v);
        addEdge(u,v);
    }
    for(int i=1;i<=n;++i) 
    { //依次以所有没被遍历过的点为根尝试搜索强连通分量 
        if(!dfn[i])
            tarjan(i);
    }
    printf("%d",ans); 
    return 0;
}