nyoj_20吝啬的国度（dfs搜索父节点）

最新推荐文章于 2021-05-19 19:41:36 发布

原创最新推荐文章于 2021-05-19 19:41:36 发布 · 240 阅读

0 ·

CC 4.0 BY-SA版权

算法与数据结构专栏收录该内容

132 篇文章

订阅专栏

本文介绍了一种解决吝啬国度问题的方法，通过DFS深度优先搜索算法和STL_vector的应用，实现从起点到各城市的必经之路查找。适用于城市间的路径规划问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题 A: 吝啬的国度

时间限制: 1 Sec 内存限制: 64 MB
提交: 62 解决: 25
[ 提交][ 状态][ 讨论版]

nyoj_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）

样例输入

样例输出

-1 1 10 10 9 8 3 1 1 8

提示

nyoj20

------------------------------------------------------------------------------------------------------------------------------------

这道题吧，写过好几遍了，还是忘了该怎么写。看了以前写的代码，才想起来。

方法：dfs + stl_vector ，不得不说vector真的很好用，但是用的次数太少了，老是忘记用法。

把结点连起来，可以看成是一棵树，定义一个par[N]，来存搜索得到的双亲结点，dfs可以一搜到底，再加上递归调用，“重孙子”都能搜出来。

注意：细节问题。一定要初始化。输入时n个结点，n-1条路。vector基本用法要掌握。

#include<stdio.h>
#include<vector>
#include<string.h>
using namespace std;
#define N 100005
int par[N];
vector<int>v[N];
void dfs(int x)
{
	for(int i=0;i<v[x].size();i++)
	{
		if(par[v[x][i]])continue;
		par[v[x][i]]=x;
		dfs(v[x][i]);
	}
}
int main()
{
	int t,n,s,a,b,i;
	scanf("%d",&t);
	while(t--)
	{
		memset(v,0,sizeof(v));
		memset(par,0,sizeof(par));
		scanf("%d%d",&n,&s);
		par[s]=-1;
		for(i=1;i<n;i++)
		{
			scanf("%d%d",&a,&b);
			v[a].push_back(b);
			v[b].push_back(a);
		}
		dfs(s);
		for(i=1;i<=n;i++)
		{
	        if(i==n)
	        printf("%d\n",par[i]);
	        else
			printf("%d ",par[i]);
		}
	}
 }