DFS-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)
样例输入
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

一开始没认真读题,想的是不是最小生成树,添加一条边的时候决策出父节点,但是在认真读题后简直打脸,只有n-1条边而已,直接从起点深搜就可以了,由于N比较大,到了10^6级,所以用vector动态分配存一下邻接表,时间可能消耗的就多了一点。
深搜是递归的,每次将邻接表中有的点有访问一次,看看有没有没有父节点的,将其父节点设为当前节点(因为深度关系,当前节点一定是其以起点为根的父节点)。
让我个人觉得奇怪的是,这道题每组数据之间居然不打换行,害得我看了几次输出格式,确认后才提交。

//
// main.cpp
// NYOJ-20-吝啬的国度
//
// Created by 袁子涵 on 15/11/24.
// Copyright © 2015年 袁子涵. All rights reserved.
//
// 324ms 5164KB

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <algorithm>
#include <vector>
#define MAXIN 100100

using namespace std;
long long int N,S;
long long int father[MAXIN];
vector<long long int>v[MAXIN];

void dfs(long long int pre)
{
    for (int i=0; i<v[pre].size(); i++) {
        if (father[v[pre][i]]) {
            continue;
        }
        father[v[pre][i]]=pre;
        dfs(v[pre][i]);
    }
}
int main(int argc, const char * argv[]) {
    int M,a,b;
    cin >> M;
    while (M--) {
        cin >> N >> S;
        memset(v, 0, sizeof(v));
        memset(father, 0, sizeof(father));
        father[S]=-1;
        for (long long int i=1; i<N; i++) {
            cin >> a >> b;
            v[a].push_back(b);
            v[b].push_back(a);
        }
        dfs(S);
        for (long long int i=1; i<=N; i++) {
            printf("%lld ",father[i]);
        }
    }
    return 0;
}

你可能感兴趣的:(c,DFS)