NYIST OJ 题目20 吝啬的王国

DFS水题。题意说明了这是一颗树，那么只要按照根节点DFS下去就好了，DFS的时候记录一下当前在哪个结点，还有父节点是谁，就AC了！

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;

const int maxn = 100000 + 10;
vector<int>ljb[maxn];
int anss[maxn], flag[maxn];
int n, s, u, v;

void dfs(int noww, int father)
{
    int i;
    anss[noww] = father;
    for (i = 0; i < ljb[noww].size(); i++)
    {
        if (flag[ljb[noww][i]] == 0)
        {
            flag[ljb[noww][i]] = 1;
            dfs(ljb[noww][i], noww);
        }
    }
}

int main()
{
    int T;
    scanf("%d", &T);
    while (T--)
    {
        int i;
        scanf("%d%d", &n, &s);
        for (i = 0; i <= n; i++) ljb[i].clear();
        memset(flag, 0, sizeof(flag));
        for (i = 1; i <= n - 1; i++)
        {
            scanf("%d%d", &u, &v);
            ljb[u].push_back(v);
            ljb[v].push_back(u);
        }
        flag[s] = 1;
        dfs(s, -1);
        for (i = 1; i <= n; i++)
        {
            if (i < n) printf("%d ", anss[i]);
            else printf("%d\n", anss[i]);
        }
    }
    return 0;
}