无根树转有根树

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1000;
int n,p[MAXN];
vector<int>G[MAXN];

void dfs(int u,int fa)
{
    int d = G[u].size();
    for(int i=0;i<d;i++)
    {
        int v=G[u][i];
        if(fa != v) dfs(v,p[v]=u);//注意判断父节点
    }
}
int main()
{
    cin >>  n;
    for(int i=0;i<n-1;i++)
    {
        int u,v;
        cin >> u >>v ;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    int root;
    cin >> root;
    p[root] = -1;
    dfs(root,-1);
    for(int i=0;i<n;i++)
    {
        cout << p[i] <<endl;
    }

    return 0;
}