CCF CSP认证201509-4 高速公路（tarjan算法求强连通分量）

201509-4 高速公路

题目

思路

求强连通分量，不会tarjan算法的我用了三次bfs，混了90分orz。
第一次去掉入度为0的点及所在链，第二次去掉出度为0的点及所在链，这样就只剩下一些环了。
每个环的便利城市对数目等于cnt*（cnt-1）/2，由此进行第三次bfs。

这么做的话，有一种情况是会出错的，当两个环有单向连接的时候可能会把两个强连通分量当成一个来计算，导致结果错误。刚开始在纠结为什么才90分，现在想的是，啊，这**竟然有90分！

于是乎我学习了一下tarjan，ac代码在下方。

90分代码如下

#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
const int maxn=10005;
vector<int> to[maxn],from[maxn];
int inDegree[maxn]={0},outDegree[maxn]={0};
bool vis[maxn];

void bfs1(int x){//剔除入度为0的点及其后的一条链
    queue<int> q;
    q.push(x);
    while(q.size()){
        int t=q.front();
        q.pop();
        vis[t]=true;
        for(int i=0;i<to[t].size();i++){
            int tt=to[t][i];
            inDegree[tt]--;
            if(inDegree[tt]==0) q.push(tt);
        }
    }
}

void bfs2(int x){//剔除出度为0的点及其前面的一条链
    queue<int> q;
    q.push(x);
    while(q.size()){
        int t=q.front();
        q.pop();
        vis[t]=true;
        for(int i=0;i<from[t].size();i++){
            int tt=from[t][i];
            outDegree[tt]--;
            if(outDegree[tt]==0) q.push(tt);
        }
    }
}

int bfs3(int x){//统计各强连通分量的点数
    int cnt=0;
    queue<int> q;
    q.push(x);
    vis[x]=true;
    while(q.size()){
        int t=q.front();
        q.pop();
        cnt++;
        for(int i=0;i<to[t].size();i++){
            int tt=to[t][i];
            if(vis[tt]==false) {q.push(tt);vis[tt]=true;}
        }
    }
    return cnt*(cnt-1)/2;
}

int main(){
    int n,m,a,b,ans=0;
    scanf("%d%d",&n,&m);
    while(m--){
        scanf("%d%d",&a,&b);
        to[a].push_back(b);
        from[b].push_back(a);
        inDegree[b]++;
        outDegree[a]++;
    }
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++){
        if(vis[i]) continue;
        if(inDegree[i]==0) bfs1(i);
    }
    for(int i=1;i<=n;i++){
        if(vis[i]) continue;
        if(outDegree[i]==0) bfs2(i);
    }
    for(int i=1;i<=n;i++){
        if(vis[i]) continue;
        ans+=bfs3(i);
    }
    printf("%d\n",ans);
    return 0;
}

tarjan算法AC代码如下

#include<cstdio>
#include<stack>
#include<vector>
using namespace std;
const int maxn=10005;
vector<int> g[maxn];
stack<int> s;
int n,dfn[maxn]={0},low[maxn]={0},time=0,ans=0;
bool ins[maxn]={false};

void tarjan(int x){
    dfn[x]=low[x]=++time;//dfn表示该点被访问的时间，low表示该点所在连通分量最早被访问的时间
    s.push(x);
    ins[x]=true;
    for(int i=0;i<g[x].size();i++){
        int t=g[x][i];
        if(dfn[t]==0){
            tarjan(t);
            low[x]=min(low[x],low[t]);
        }
        else if(ins[t]){//在栈中，说明属于同一强连通分量
            low[x]=min(low[x],dfn[t]);
        }
    }
    int temp,cnt=0;
    if(low[x]==dfn[x]){//x是该强连通分量中第一个被访问的点
        while(!s.empty()){
            temp=s.top();
            ins[temp]=false;
            s.pop();
            cnt++;
            if(temp==x) break;//该强连通分量已全部出栈
        }
        ans+=cnt*(cnt-1)/2;
    }
}

int main(){
    int m,a,b;
    scanf("%d%d",&n,&m);
    while(m--){
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
    }
    for(int i=1;i<=n;i++){
        if(dfn[i]==0){
            tarjan(i);
        }
    }
    printf("%d\n",ans);
    return 0;
}