图论专题小结：无根树转有根树

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

2.思路：用vector数组存放每个结点的相邻结点，vector与n成正比。

#define N 1000000+10
vector<int>G[N];
int p[N];
int 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;
}