题目来自于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常出现的点。