洛谷 P3884 二叉树问题

题意中的两点距离需要稍微思考一下，很容易想到 $lca$，然后注意一下本题中距离的定义；深度，跑 $lca$所需要的dfs已经跑过了每个点的深度；宽度，一开始想用统计子树的方法，然后发现递归可能不太好写 我写不出来，突然发现直接桶排一下就行了，而数据也非常适合桶排。
数据较小，所以 $lca$中的 $i$ 其实开到8就行，个人写的9

#include<bits/stdc++.h>
using namespace std;
int n;
int dep_max,wid_max;//存储
vector<int> e[107];//存边
int dep[107],fa[107][10];//lca
int tim[107];//桶排数组，存每种深度出现的次数
void dfs(int fro,int fat){
	dep[fro]=dep[fat]+1;
	++tim[dep[fro]];
	dep_max=max(dep_max,dep[fro]);
	fa[fro][0]=fat;
	for(int i=1;i<10;++i)
		fa[fro][i]=fa[fa[fro][i-1]][i-1];
	for(int i=0;i<e[fro].size();++i){
		int y=e[fro][i];
		if(y==fat)continue;
		dfs(y,fro);
	}
	return;
}
int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=9;i>=0;--i)
	if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
	if(x==y)return x;
	for(int i=9;i>=0;--i)
	if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int dis(int x,int y){
	int k=lca(x,y);
	return 2*(dep[x]-dep[k])+dep[y]-dep[k];//注意：本题中，x->y的距离与y->x的距离是不一样的！
}
void width(){//桶排
	for(int i=1;i<=dep_max;++i)
	wid_max=max(wid_max,tim[i]);
	cout<<wid_max<<endl;
	return;
}
int main(){
	cin>>n;
	for(int i=1;i<n;++i){//建图
		int x,y;cin>>x>>y;
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs(1,0);
	cout<<dep_max<<endl;
	width();
	int u,v;cin>>u>>v;
	cout<<dis(u,v);
	return 0;
}