吝啬的国度
时间限制:1000 ms | 内存限制:65535 KB
难度:3
- 描述
在一个吝啬的国度里有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[N][N],根本没办法编译,==!于是放弃邻接矩阵,改用邻接表存储,我把代码那个写啊。。。。指针,结构体,头结点什么的一会儿就晕了, 后来看到有人使用STL中的 vector ,可以用来代替二维数组,不过差不多,于是我就试着用vector (第一次用也不知道理解的对不对,,以前见过没用), vector <int> city[N]; 相当于N个vector,跟二维数组差不多
代码:
#include<iostream> #include<cstdio> #include<cstring> #include<vector> #define N 100005 using namespace std; vector <int> city[N]; int p[N]; bool vis[N]; 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[n]); } 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; }