【图论入门】吝啬的国度

最新推荐文章于 2018-05-19 14:14:03 发布

原创最新推荐文章于 2018-05-19 14:14:03 发布 · 556 阅读

0 ·

CC 4.0 BY-SA版权

算法入门专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一个典型的图论问题：使用广度优先搜索（BFS）算法来确定从起点到任意其他点的路径。通过具体实例展示了如何利用队列和邻接表实现算法，并分享了调试过程中的常见错误。

题目来自于nyist第20题

描述
在一个吝啬的国度里有N个城市，这N个城市间只有N-1条路把这个N个城市连接起来。现在，Tom在第S号城市，他有张该国地图，他想知道如果自己要去参观第T号城市，必须经过的前一个城市是几号城市（假设你不走重复的路）。
输入
第一行输入一个整数M表示测试数据共有M(1<=M<=5)组
每组测试数据的第一行输入一个正整数N(1<=N<=100000)和一个正整数S(1<=S<=100000)，N表示城市的总个数，S表示参观者所在城市的编号
随后的N-1行，每行有两个正整数a,b(1<=a,b<=N)，表示第a号城市和第b号城市之间有一条路连通。
输出
每组测试数据输N个正整数，其中，第i个数表示从S走到i号城市，必须要经过的上一个城市的编号。（其中i=S时，请输出-1）

本题中N个节点只有N-1条边，由树的性质，可知图中没有环路，采用DFS或者BFS都可以，这里采用BFS（因为写的多了顺手。。），采用queue以及vector维护数据，题目较为简单，这里直接贴出代码。

#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#define Max 100000
using namespace std;
queue<int> Q;

int via[Max+5];
int path[Max+5];
int main()
{
	int M;
	int N,S,i;
	int a,b,num,temp;
	scanf("%d",&M);
	while(M--)
	{
		vector<int> G[Max+5];   //开始定义在全局中，一直WA，后来才发现对每组数据没有初始化
		memset(via,0,sizeof(via));
		memset(path,-1,sizeof(path));
		num = 0;temp = 0;
		scanf("%d %d",&N,&S);
		for(i=0;i <N-1;i++)
		{
			scanf("%d %d",&a,&b);
			G[a].push_back(b);

			G[b].push_back(a);
		}
		Q.push(S);
		via[S] = 1;
		while(!Q.empty())
		{
			temp = Q.front();Q.pop();
			for(i = 0;i < G[temp].size();i++)
			{
				if(via[G[temp][i]] != 1)
				{
					Q.push(G[temp][i]);
					path[G[temp][i]] = temp;
					via[G[temp][i]] = 1;
				}
			}
		}
		for(i=1;i <=N;i++)
			printf("%d ",path[i]);
	}
	return 0; 
}

这题较为简单，主要是前段时间各大公司笔试题做懵逼了，做了一段时间DP之后，图论题太生疏了，基础题也花了很久时间调bug，写下这篇提醒自己要多方面练习，总结一下bug常出现的点。