时间限制:1秒 内存限制:128M
题目描述
求树的各个节点的深度可以自上而下统计信息。起初,我们已知根节点的的深度为0.若节点x的深度为d[x],则它的子节点y的深度为d[y]=d[x]+1,现在给你一棵树,让你输出每个节点的深度。
输入描述
第一行,两个整数n(1<=n<=200000),s,n表示树的节点的个数,s表示树的根节点的编号。
接下来的n-1行中,每行有两个整数x,y,表示x和y有一条边。
输出描述
输出占一行,n个数 ,第i个数表示节点i的深度。
样例
输入
9 1 1 2 1 7 1 4 2 8 2 5 4 6 4 3 3 9
输出
0 1 2 1 2 2 1 2 3
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
int n,s;
int x,y;
const int N=1e6+10;
int head[N*2],vis[N*2],ver[N*2],Next[N*2],step[2*N],tot=-1,cnt;
void Add(int x,int y) {
ver[++tot]=y;
Next[tot]=head[x];
head[x]=tot;
}
void dfs(int x) {
vis[x]=1;
for(int i=head[x]; i!=-1; i=Next[i]) {
int ljd=ver[i];
if(vis[ljd]==0) {
step[ljd]=step[x]+1;
dfs(ljd);
}
}
}
int main() {
memset(head,-1,sizeof(head));
cin>>n>>s;
for(int i=1; i<=n; i++) {
cin>>x>>y;
Add(x,y);
Add(y,x);
}
step[s]=0;
dfs(s);
for(int i=1; i<=n; i++) {
cout<<step[i]<<" ";
}
return 0;
}