树上基本统计（模板题）

输入一棵树，1是根节点。求每个节点的父节点、深度、子树大小、子树高度

代码：

#include <bits/stdc++.h>
using namespace std;
const int N=100009;
vector<int> to[N];
int d[N],p[N],sz[N],h[N],n;
void add(int u,int v){
	to[u].push_back(v);
	to[v].push_back(u);
}
void input(){
	cin>>n;
	for(int i=1;i<=n-1;i++){
		int u,v;
		cin>>u>>v;
		add(u,v);
	}
}
void dfs(int u,int fa){
	p[u]=fa;
	d[u]=d[fa]+1;
	sz[u]=1;
	h[u]=1;
	for(int i=0;i<to[u].size();i++){
		int v=to[u][i];
		if(v==fa) continue;
		dfs(v,u);
		sz[u]+=sz[v];
		h[u]=max(h[u],h[v]+1);
	}
}
int main(){
	input();
	dfs(1,0);
	for(int i=1;i<=n;i++,cout<<endl) cout<<p[i]<<" "<<d[i]<<" "<<sz[i]<<" "<<h[i];
	return 0;
}