【算法学习】强连通分量

今天意外的发现了强连通分量这个神奇的东西。任何一个图的强连通分量，必定是原图的深度优先搜索树的子树。

#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
using namespace std;
const int maxn=10000;
struct Edge
{
	int from,to,d;
};
vector<Edge> edges;
vector<int> G[maxn];
	int n,m;
int flag[maxn];
int color[maxn];
stack<int> s;
int low[maxn];
int index[maxn];

void DFS(int x,int &sig,int &count)
{
	s.push(x);
	flag[x]=1;
	low[x]=index[x]=++sig;
	for(int i=0;i<G[x].size();i++)
	{
		Edge e=edges[G[x][i]];
		int v=e.to;
		if(flag[v]==0)
		{
			DFS(v,sig,count);
			if(low[v]<low[x])	low[x]=low[v];
		}
		else if(flag[v]==1&&index[v]<low[x])
			low[x]=index[v];
	}
	if(low[x]==index[x])
	{
		count++;
		int t;
		do
		{
			t=s.top(); s.pop();
			color[t]=count;
			flag[t]=2;
		}while(t!=x);
	}
}

int Tarjan()
{
	int sig,count;//count为强连通分量数量,signal为时间戳
	memset(flag,0,sizeof(flag));
	sig=count=0;
	for(int i=1;i<=n;i++)
	{
		if(flag[i]==0)
			DFS(i,sig,count);
	}
	return count;
}

int main(void)
{

	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)
	{
		int x,y,z,m=edges.size();
		scanf("%d%d%d",&x,&y,&z);
		G[x].push_back(m);
		edges.push_back((Edge){x,y,z});
	}
	printf("%d\n",Tarjan());
	
	return 0;
}