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

分析：

所谓的不走重复的路，就是要求从这个起始点开始走，第一次到达这个点的时候肯定是没有走过重复路径的。我们可以用一个数组来表示这些点的前一个点，刚开始的时候都是0，然后一旦有路径经过它，就给他赋值为前一个点。

代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
#define INF 0x3f3f3f3f
vector<int >v[100009];
int en[100009];
int N,S;
void init()///数据初始化
{
    memset(v,0,sizeof(v));
    memset(en,0,sizeof(en));
}

void bfs(int S)
{
    queue<int>q;
    q.push(S);
    int now;
    while(!q.empty())
    {
        now=q.front();
        q.pop();
        for(int i=0; i<v[now].size(); i++)
        {
            if(en[v[now][i]]==0)///也就意味着这个点没有走过呢
            {
                en[v[now][i]]=now;///把它的上一个节点记录下来
                q.push(v[now][i]);
            }
        }
    }
}
int main()
{
    int M,N1;
    scanf("%d",&M);
    while(M--)
    {
        init();
        scanf("%d%d",&N,&S);
        N1=N;
        N=N-1;
        int a,b;
        while(N--)
        {
            scanf("%d%d",&a,&b);
            v[a].push_back(b);
            v[b].push_back(a);
        }
        en[S]=-1;
        bfs(S);

        for(int i=1; i<=N1; i++)
        {
            printf("%d ",en[i]);
        }
        printf("
");
    }
    return 0;
}