Tarjan算法模板(求有向图的连通分量)

 

#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
#define maxn 500//图的规模
using namespace std;

vector<int> G[maxn];//邻接表存图
int pre[maxn], lowlink[maxn], sccno[maxn], dfs_clock, scc_cnt;
stack<int> S;

void dfs(int u){
       pre[u] = lowlink[u] = ++dfs_clock;
       S.push(u);
       for(int i=0; i < G[u].size(); i++){
              int v = G[u][i];
              if(!pre[v]){
                     dfs(v);
                     lowlink[u] = min(lowlink[u], lowlink[v]);
              }else if(!sccno[v]) {
                     lowlink[u] = min(lowlink[u], pre[v]);
              }
       }
       if(lowlink[u] == pre[u]) {
              scc_cnt ++ ;
              for(;;) {
                     int  x = S.top(); S.pop();
                     sccno[x] = scc_cnt;
                     if(x == u) break;
              }
       }
}

void find_scc(int n){
       dfs_clock = scc_cnt = 0;
       memset(sccno, 0, sizeof(sccno));
       memset(pre, 0, sizeof(pre));
       for(int i = 0; i < n; i++)
              if(!pre[i]) dfs(i);
}
int main()
{
       int T,i;
       scanf("%d",&T);
       while(T--)
       {
              int n,m;
              scanf("%d%d",&n,&m);
              for(i=0;i<m;i++)
              {
                     int a,b;
                     scanf("%d%d",&a,&b);
                     G[a].push_back(b);
              }
              find_scc(n);
              for(i=0;i<n;i++)
              {
                     printf("%d ",sccno[i]);
              }
              printf("\n");
       }
	return 0;
}

 

     图用 邻接表存储

sccno[i]表示  i所在的连通分量数
scc_cnt       表示一共的连通分量数

 

 

 

 

 

你可能感兴趣的:(tar)