DFS遍历图代码(算法导论思路实现)

#include<iostream>
#include<vector>
#include<cstring>
#include<cstdio>
using namespace std;

const int maxn = 10005;
int color[maxn]; //0代表白色,1代表灰色,2代表黑色
vector<int> vec[maxn]; //邻接表 
int step; //全局变量代表总步数
int stime[maxn]; //每个点遍历的时候的起始时间
int etime[maxn]; //每个点遍历的时候的终止时间 
bool exist[maxn]; //判断图中是否存在该点 
int parent[maxn];

void init()
{
	memset(color,0,sizeof(color)); //初始化都为白色
	memset(stime,0,sizeof(stime));
	memset(etime,0,sizeof(etime));
	memset(parent,-1,sizeof(parent)); //初始化每个结点的父亲都为-1 
	step = 0;
}

void dfs_visit(int u)
{
	int i;
	int pos;
	step++;
	stime[u] = step;
	color[u] = 1; //刚刚访问到这一结点，置为灰色 
	int len = vec[u].size();
	for(i=0;i<len;i++)
	{
		pos = vec[u][i];
		if(color[pos] == 0)
		{
			parent[pos] = u;
			dfs_visit(pos);
		}
	}
	color[u] = 2; //邻接点全部遍历完了，置为黑色 
	step++;
	etime[u] = step;
}

void dfs()
{
	int i;
	for(i=1;i<maxn;i++)
	{
		if(exist[i] != false && color[i] == 0) //存在这个点且这个点为白色 
		{
			dfs_visit(i);
		}
	}
}

void print()
{
	int i;
	for(i=0;i<maxn;i++)
	{
		if(exist[i] != false)
		{
			cout<<i<<"点的信息如下:"<<endl;
			cout<<"起始时间:"<<" "<<stime[i]<<" "<<"结束时间:"<<" "<<etime[i]<<endl; 
		}
	}
}

int main()
{
	int start;
	int num1,num2;
	init();
	freopen("1.txt","r",stdin);
	while(cin>>num1>>num2 && num1 != 0 && num2 != 0)
	{
		vec[num1].push_back(num2); //有向图 
		exist[num1] = true;
		exist[num2] = true;
	}
	/*for(int i=0;i<maxn;i++)
	{
		if(exist[i] == true)
		{
			cout<<i<<endl;
		}
	}*/
	dfs();
	print();
	return 0;	
}