nyoj-20-吝啬的国度(深搜)

吝啬的国度

时间限制：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）

样例输入

110 11 91 88 1010 38 61 210 49 53 7

样例输出

-1 1 10 10 9 8 3 1 1 8

解题思路：

      一开始用了队列来求，起点入队，然后出队，找到跟起点相连的城市进队，那么这些城市的前一步就是出队的城市了，然后开始不停地出队入队，直到队空为止。思路不错，但是超时了。

      经过看大神的代码，发现这个题可以用深搜来解决，再一看题目发现这确实符合深搜的特性。

     不管是队列还是深搜，都需要标记。

     队列用到了#include<queue>STL函数，

#include <iostream>#include <queue>using namespace std; //这几个头文件必不可少int main(){queue<类型（如int）> q; //使用前需定义一个queue变量，且定义时已经初始化while(!q.empty()) q.pop(); //重复使用时，用这个初始化（空则返回1，不空返回0）q.push(1); //进队列q.pop(); //出队列int v=q.front(); //得到队首的值int s=q.size(); //得到队列里元素个数return 0;}

      因为给定的城市N的数目太大，建立数组需要用到#include<vector>，vector就是一个不定长数组，vector<int>a就是一个类似于int a[]的整数数组，只不过他的长度不确定，可以用a.size()读取他的长度。

      而vector<int>a[max]就是一个二维数组，只是第一维的大小是固定的（不超过max），二维的大小就不固定了，这道题之所以用到vector就是利用了他的不定长，如果直接建立二维数组a[n][n]，n太大了，这样的二维数组绝对超出内存。

      (1)头文件#include<vector>.

      (2)创建vector对象，vector<int> vec;

      (3)尾部插入数字：vec.push_back(a);

      (4)使用下标访问元素，cout<<vec[0]<<endl;记住下标是从0开始的。

      (5)向量大小:vec.size();

代码：

//队列思路（超时） #include<stdio.h>#include<string.h>#include<iostream>#include<queue>using namespace std;int sta[110000];int map[110000][3];int ans[110000];int main(){int m;int n,s;int i,j,k;int now;queue<int>q;scanf("%d",&m);while(m--){scanf("%d%d",&n,&s);q.push(s);for(i=1;i<n;i++){map[i][0]=1;    scanf("%d%d",&map[i][1],&map[i][2]);}memset(ans,0,sizeof(ans));ans[s]=-1;while(!q.empty()){now=q.front();q.pop();for(i=1;i<n;i++){if(map[i][0]&&map[i][1]==now){    q.push(map[i][2]);    ans[map[i][2]]=now;    map[i][0]=0;}else if(map[i][0]&&map[i][2]==now){    q.push(map[i][1]);    ans[map[i][1]]=now;    map[i][0]=0;}}}for(i=1;i<=n;i++){printf("%d",ans[i]);if(i!=n)    printf(" ");else    printf("\n");}}return 0;}

//深搜代码（AC） #include<stdio.h>#include<string.h>#include<iostream>#include<vector>#include<algorithm>using namespace std;int pre[100005];vector<int>v[100005];void dfs(int s){int i;for(i=0;i<v[s].size();i++){if(pre[v[s][i]])    continue;pre[v[s][i]]=s;dfs(v[s][i]);}}int main(){int m;int n,s;int a,b;int i,j;scanf("%d",&m);while(m--){memset(v,0,sizeof(v));memset(pre,0,sizeof(pre));scanf("%d%d",&n,&s);pre[s]=-1;for(i=0;i<n-1;i++){scanf("%d%d",&a,&b);v[a].push_back(b);v[b].push_back(a);}dfs(s);for(i=1;i<=n;i++){    printf("%d",pre[i]);    if(i!=n)        printf(" ");    else        printf("\n");}}return 0;}