记录编号 463453 评测结果 AAAAAA
题目名称 无根树转有根树 最终得分 100
用户昵称 Gravatar东林桂香 是否通过 通过
代码语言 C++ 运行时间 0.667 s
提交时间 2017-10-24 10:02:18 内存使用 19.39 MiB
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<vector>
#define maxn 1000005
using namespace std;
int n,m,root,fa[maxn],ans[maxn];
vector<int>ha[maxn];
void dfs(int now,int fat){
	int pos=ha[now].size();
	for(int i=0;i<pos;i++){
		if(ha[now][i]!=fat){
			fa[ha[now][i]]=now;
			dfs(ha[now][i],now);
		}
	}
} 
int main(){
	freopen("wgs.in","r",stdin);
	freopen("wgs.out","w",stdout);
	scanf("%d%d",&n,&root);
	for(int i=0;i<=n;i++){
		fa[i]=i;
		if(i==root)fa[i]=-1;
	}
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		ha[x].push_back(y);
		ha[y].push_back(x);
	}
	scanf("%d",&m);
	for(int i=1;i<=m;i++)scanf("%d",&ans[i]);
	dfs(root,-1);
	for(int i=1;i<=m;i++){
		printf("%d ",fa[ans[i]]);
	}
	return 0;
}