文章标题CCF201509-4高速公路

问题描述
　　某国有n个城市，为了使得城市间的交通更便利，该国国王打算在城市之间修一些高速公路，由于经费限制，国王打算第一阶段先在部分城市之间修一些单向的高速公路。
　　现在，大臣们帮国王拟了一个修高速公路的计划。看了计划后，国王发现，有些城市之间可以通过高速公路直接（不经过其他城市）或间接（经过一个或多个其他城市）到达，而有的却不能。如果城市A可以通过高速公路到达城市B，而且城市B也可以通过高速公路到达城市A，则这两个城市被称为便利城市对。
　　国王想知道，在大臣们给他的计划中，有多少个便利城市对。
输入格式
　　输入的第一行包含两个整数n, m，分别表示城市和单向高速公路的数量。
　　接下来m行，每行两个整数a, b，表示城市a有一条单向的高速公路连向城市b。
输出格式
　　输出一行，包含一个整数，表示便利城市对的数量。
样例输入
5 5
1 2
2 3
3 4
4 2
3 5
样例输出
3
样例说明

　　城市间的连接如图所示。有3个便利城市对，它们分别是(2, 3), (2, 4), (3, 4)，请注意(2, 3)和(3, 2)看成同一个便利城市对。
评测用例规模与约定
　　前30%的评测用例满足1 ≤ n ≤ 100, 1 ≤ m ≤ 1000；
　　前60%的评测用例满足1 ≤ n ≤ 1000, 1 ≤ m ≤ 10000；
　　所有评测用例满足1 ≤ n ≤ 10000, 1 ≤ m ≤ 100000。

//有向图强连通分量 -- tarjin算法
#include<iostream>
#include<stack>
#include<vector>
#include<cstring>
#include<algorithm>
#include<fstream>
#define N 100005

using namespace std;

int n, m;               //城市和单向高速公路的数量
int DFN[N];             //节点u搜索的次序编号(时间戳)
int Low[N];             //u或u的子树能够追溯到的最早的栈中节点的次序号
bool vis[N];            //是否已访问 
bool inS[N];            //是否在栈中 
int time = 0;           //时间戳  
vector <int> G[N];      //邻接表 
stack <int> s;          //DFS栈 
int result = 0;         //最终结果：所有强连通分量中结点对的个数 

void tarjan( int u ){
    DFN[u] = Low[u] = ++time;       //时间戳从1开始 
    s.push(u);                      //结点u入栈
    inS[u] = 1;                     //在栈中
    vis[u] = 1;                     //已访问 
    int v;
    for( int i = 0; i < G[u].size(); i++ ){ //访问邻接表 
        v = G[u][i];
        if( vis[v] == 0 ){
            tarjan( v );
            Low[u] = min( Low[u], Low[v] );
        }
        else if( inS[v] == 1 ){
            Low[u] = min( Low[u], DFN[v] ); 
        }
    }   
    int count = 0;          //强连通分量中结点个数
    if( DFN[u] == Low[u] ){
        do{
            v = s.top();
            inS[v] = 0;
            s.pop();
            count++;
        }while( u != v );

        if( count > 0 ){
            result += count * ( count-1 ) / 2;
        } 
    }

}

int main(){
    int a, b;

    memset( DFN, 0, sizeof(DFN) );
    memset( Low, 0, sizeof(Low) ); 
    memset( G, 0, sizeof(G) ); 
    memset( inS, 0, sizeof(inS) ); 
    memset( vis, 0, sizeof(vis) ); 

//  ifstream fin("00.txt", ios::in);
//  fin >> n >> m;
    cin >> n >> m;

    for( int i = 0; i < m; i++ ){
//      fin >> a >> b;
        cin >> a >> b;
        G[a].push_back(b) ;
    }

    for( int i = 1; i <= n; i++ ){
        if( vis[i] == 0 ){
            tarjan( i );
        }
    }

    cout << result; 

    return 0;
}