NYOJ 题目20吝啬的国度（DFS）

吝啬的国度

时间限制：1000 ms  |  内存限制：65535 KB

[align=center]难度：3[/align]

            描述
在一个吝啬的国度里有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

首先想到的是使用邻接矩阵（菜鸟！没办法o(╯□╰)o）， 差不多快写完了，然后看题目中N的范围，100000，开一个二维数组city

,根本没办法编译，==！于是放弃邻接矩阵，改用邻接表存储，我把代码那个写啊。。。。指针，结构体，头结点什么的一会儿就晕了， 后来看到有人使用STL中的 vector ，可以用来代替二维数组，不过差不多，于是我就试着用vector
（第一次用也不知道理解的对不对，，以前见过没用）， vector <int> city
;  相当于N个vector，跟二维数组差不多

 

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#define N 100005
using namespace std;

vector <int> city
;
int p
;
bool vis
;

void DFS(int x)
{
vis[x] = 1;
int i=0,len;
len = city[x].size();
for(i=0;i<len;i++)
{
if(!vis[city[x][i]])
{
p[city[x][i]]=x;
DFS(city[x][i]);
}
//	i++;
}
}
int main()
{
int m,n,s,i,x,y;

scanf("%d",&m);

while(m--)
{
scanf("%d%d",&n,&s);
for(i=0;i<=n;i++) city[i].clear();//这里初始化清空，第一次没写WA了
for(i=0;i<n-1;i++)
{
scanf("%d%d",&x,&y);
city[x].push_back(y);
city[y].push_back(x);
}
memset(vis,0,sizeof(vis));
p[s]=-1;
DFS(s);
for(i=1;i<n;i++) printf("%d ",p[i]); printf("%d\n",p
);
}
return 0;
}

 

别人优秀代码，用的map，还没用过，改天研究研究！

#include <stdio.h>
#include <memory.h>

int map[100005];

void Adjust(int currentCity)
{
int priorCity = map[currentCity];
if (priorCity != 0)
{
Adjust(priorCity);
map[priorCity] = currentCity;
}
}

int main()
{
int i, testNum, cityNum, startCity, cityA, cityB;
scanf("%d", &testNum);
while (testNum-- != 0)
{
scanf("%d%d", &cityNum, &startCity);
memset(map, 0, sizeof(int)*cityNum + 1);
for (i = 1; i < cityNum; i++)
{
scanf("%d%d", &cityA, &cityB);
if (map[cityB] == 0)
{
map[cityB] = cityA;
}
else
{
Adjust(cityA);
map[cityA] = cityB;
}
}
Adjust(startCity);
map[startCity] = - 1;
for (i = 1; i < cityNum; i++)
{
printf("%d ", map[i]);
}
printf("%d\n", map[i]);
}
return 0;
}