NYOJ-20题-经典搜索（vector）

本文介绍了一种利用深度优先搜索(DFS)算法解决图论问题的方法，通过具体实例展示了如何构建图的数据结构并实现DFS算法来寻找从起始点到各顶点的路径。

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

题目链接：http://acm.nyist.edu.cn/JudgeOnline/problem.php?pid=20

#include<iostream>

#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>
#include<stack>
#include<vector>
#include<algorithm>
#define N 100005
using namespace std;

int ans[N],flag[N],n,s;//n代表城市的个数 ,s代表出发城市编号
vector<int> map[N];//数据太大，用不了二维数组

void init()
{
for(int i = 1; i <= n; i++)
map[i].clear();
memset(ans,-1,sizeof(ans));
memset(flag,0,sizeof(flag));
}

void DFS(int s)
{
flag[s] = 1;
//由于vector清空了，所以里面的size为0，不能使用map[s][i]
// for(int i = 1; i <= n; i++)
// {
// if(map[s][i] == 1 && !flag[i])
// {
// ans[i] = s;
// DFS(i);
// }
// }
//尽量用迭代器的方式遍历
for(vector<int>::iterator it=map[s].begin();it != map[s].end();it++){
if(!flag[*it]){
ans[*it] = s;
DFS(*it);
}
}
}

int main()
{
int t,from,to;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&s);
init();
int temp = n-1;
while(temp--)
{
scanf("%d%d",&from,&to);
map[from].push_back(to);
map[to].push_back(from);
//无向图
}
DFS(s);
//for_each(ans,ans+n,print());
for(int i = 1; i <= n; i++)
{
if(i == n)
printf("%d\n",ans[i]);
else
printf("%d ",ans[i]);
}
}
return 0;
}