无根树转为有根树-vector

输入一个n个结点的无根树的各条边，并指定一个根结点，要求把该树转化为有根树，输出各个结点的父节点编号。n<=10^6.

 //输入无根树的结点个数n，输入n-1条边(u, v)，输入欲指定的根的编号root，建立以root为根的树  
/*可变长数组 一个有maxn行 当时每行长度可以不同的数组 

      用于表示树中结点间的关系  */

主程序中设置p[root]=-1(表示根结点不存在)，然后调用dfs(root,-1)即可。如果忘了判断v是否和其父结点相等，就会引起无限循环。

/* 
  输出格式  p[该节点] =  该节点父亲 
  例如 p[5] = 1; 结点5的父亲是1;  
  当该节点为根节点 则对应的为-1  
*/ 

/*终端：

输入：

8

0 1 

0 2

0 3

1 4

1 5

5 6

5 7

输出：

1  -1  0  0  1  1  5  5

*/

各个结点所对应的父结点：

0  1   2  3   4  5  6 7 

1  -1  0  0  1  1  5  5

#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
#define N 1000000+10
vector<int>G[N];
int p[N],n;
void read_tree()
{
    int u, v;
    scanf("%d", &n);
    for (int i = 0; i < n - 1; i++)
    {
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
}
void dfs(int u, int fa)
{
    int d = G[u].size();
    for (int i = 0; i < d; i++)
    {
        int v = G[u][i];
        if (v != fa)
            dfs(v, p[v] = u);
    }
}
int main()
{
    read_tree();
    int root;
    cin >> root;
    p[root] = -1;
    dfs(root, -1);
    for (int i = 0; i < n; i++)
        printf("%d%c", p[i], i == n - 1 ? '\n' : ' ');
    return 0;
}