吝啬的国度
描述
在一个吝啬的国度里有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)
样例输入
1
10 1
1 9
1 8
8 10
10 3
8 6
1 2
10 4
9 5
3 7
样例输出
-1 1 10 10 9 8 3 1 1 8
分析:用vector建图+深搜,深搜的话就是把每个结点的父结点给记录下来,最后遍历输出即可
#include<bits/stdc++.h>
using namespace std;
#include<vector>
const int maxn = 100000 + 10;
vector<int> g[maxn];
int f[maxn];
void dfs(int x)
{
for(int i=0;i<g[x].size();i++)
{
if(f[g[x][i]]!=0)//如果有了父结点,继续下一个
{
continue;
}
f[g[x][i]]=x;//没有父结点,当前x作为它的父结点
dfs(g[x][i]);//沿着当前结点继续搜索
}
}
int main() {
int t;
int a, b;
cin >> t;
int n, m;
while(t--){
cin >> n >> m;//城市总个数,参观者所在的城市编号
for(int i=1;i<n;i++) {
cin >> a >> b;
g[a].push_back(b);
g[b].push_back(a); //vector建图
}
memset(f,0,sizeof(f));
f[m]=-1;
dfs(m);
for(int i=1;i<=n;i++)
cout<<f[i]<<" ";
cout<<endl;
}}
链式前向星+广搜
#include<bits/stdc++.h>
using namespace std;
int n, m, u, v;
int cnt;
const int maxn = 100000 + 10;
queue<int> q;
int head[maxn];
int vis[maxn];
int qianqu[maxn];
struct node {
int to;
int next;
} edge[maxn];
void init()
{
cnt=0;
memset(vis,0,sizeof(vis));
memset(qianqu,-1,sizeof(qianqu));
memset(head,-1,sizeof(head));
}
void add(int u, int v) {
edge[cnt].to = v;
edge[cnt].next = head[u];
head[u] = cnt++;
}//链式前向星建图
void bfs(int x) {
vis[x] = 1;
q.push(x);
while (!q.empty()) {
int h = q.front();
q.pop();
for (int i = head[h]; i != -1; i = edge[i].next) {
int nxt = edge[i].to; //终点
if (vis[nxt] == 0) {
q.push(nxt);
qianqu[nxt] = h;
bfs(nxt);
}
}
}
}
int main() {
int t;
cin >> t;
init();
while (t--) {
cin >> n >> m;
for (int i = 1; i < n; i++) {
cin >> u >> v;
add(u, v);
add(v, u);
}
bfs(m);
for(int i=1;i<=n;i++)
cout<<qianqu[i]<<" ";
}
}