华为机试—条条道路通罗马

标签：华为机试   条条道路通罗马   深度优先遍历   深搜   map   

题目：

基本思想：

这是一个深搜问题，不过呢，你可以看到题目截图中有答案提示：城市数目较多，因此不能使用完全遍历，无法满足时间复杂度要求。所以需要对数据做一下预处理，也就是将map转换为链表形式的，本来map应该是只有0,1的值，表示有路径或者没路径。不过为了节约点内存（虽然没什么必要），所以map复用。当前节点i如果和另外的节点j相连，则将节点j放在map中。第0个元素存放当前访问到第几个节点（从1开始，深搜回溯的时候有用），后面依次存放与之相邻的 节点，最后以-1结束。

代码：

#include<stdio.h>
#include<string.h>

int main()
{
	const int N = 10;
	int map[N][N];
	int stack[N];
	bool visited[N];

	int n,a,b;
	int Begin;
	int i,j;
	while(scanf("%d %d %d", &n, &a, &b) != EOF)
	{
		for(i = 0; i < n; ++i)
		{
			for(j = 0; j < n; ++j)
			{
				scanf("%d", &map[i][j]);
			}
		}
		for(i = 0; i < n; ++i)
		{
			int index = 0;
			map[i][0] = 1;
			for(j = 0; j < n; ++j)
			{
				if(map[i][j] && i != j)
					map[i][++index] = j;//记录下可达的点				
			}
			map[i][++index] = -1;//-1结束
		}

		//深搜
		memset(visited, 0, N * sizeof(bool));
		int top = -1;
		int ans = 0;
		stack[++top] = a;//a入栈
		visited[a] = true;
		while(top > -1)//栈不空
		{
			Begin = stack[top];//取栈顶元素
			for(i = map[Begin][0]; map[Begin][i] != -1; ++i)
			{
				if(map[Begin][i] != b && !visited[map[Begin][i]])//未访问过
				{
					stack[++top] = map[Begin][i];
					visited[map[Begin][i]] = true;
					map[Begin][0] = i + 1;
					break;
				}
				if(map[Begin][i] == b)
					++ans;
			}
			if(map[Begin][i] == -1)//这个点的所有路径都走过了，退栈
			{
				--top;
			}
		}// end for while
		
		printf("%d\n", ans);		
	}

	return 0;
}

测试结果：

华为机试—条条道路通罗马

标签：华为机试   条条道路通罗马   深度优先遍历   深搜   map   

原文地址：http://blog.csdn.net/wtyvhreal/article/details/45246159