题目20 吝啬的国度 (DFS)

题目链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=20

首先这题看出是个图的生成树，用pre数组记录下父节点的值，深搜即可，深搜过程中记录父节点位置。

由于节点树较多，这里采用C++ STL中的vector容器来存储图。

#include<stdio.h>
#include<vector>
#include<string.h>
#include<stdlib.h>

using namespace std;

int pre[100005];
vector <int> v[100005];

void dfs(int x)
{
    for(int i=0;i<v[x].size();i++)
    {
        if(!pre[v[x][i]])   //如果父节点未被记录过
        {
            pre[v[x][i]]=x;  //记录父节点位置
            dfs(v[x][i]);      //沿路深搜
        }
    }
}

int main()
{
    int T,x,y,N,s;
    scanf("%d",&T);
    while(T--)
    {
        memset(v,0,sizeof(v));
        memset(pre,0,sizeof(pre));
        scanf("%d%d",&N,&s);
        pre[s]=-1;                  //起点父节点为-1
        for(int i=0;i<N-1;i++)
        {
            scanf("%d%d",&x,&y);
            v[x].push_back(y);       //无向图存储
            v[y].push_back(x);
        }
        dfs(s);              //从起点开始搜索
        for(int i=1;i<=N;i++)
            printf("%d ",pre[i]);
    }
    return 0;
}