小米2017秋招真题：二叉树的高度

【题目描述】
现在有一棵合法的二叉树，树的节点都是用数字表示，现在给定这棵树上所有的父子关系，求这棵二叉树的高度。

【输入描述】
输入的第一行表示节点的个数n（1 ≤ n ≤ 1000，节点的编号为0到n-1）组成，下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号。

【输出描述】
输出树的高度，为一个整数。

【输入样例】
5
0 1
0 2
1 3
1 4

【输出样例】
3

【算法分析】
建议查看文章 https://blog.csdn.net/hnjzsyjyj/article/details/119938620 
它是介绍求树的高度的代码，所以更具普遍意义。

【算法代码】

#include <bits/stdc++.h>
using namespace std;

int hi(vector< vector<int> > &t, int idx) {
	int cnt=t[idx].size();

	if(cnt==0) return 1;
	if(cnt==1) return hi(t, t[idx][0])+1;
	else {
		int le=hi(t,t[idx][0]);
		int ri=hi(t,t[idx][1]);
		return le>ri?le+1:ri+1;
	}
}

int main() {
	int n;
	cin>>n;
	vector< vector<int> > tree(n); //using vector to build tree
	
	for(int i=0; i<n-1; i++) {
		int fu;
		int zi;
		cin>>fu>>zi;
		tree[fu].push_back(zi);
	}

	cout<<hi(tree,0)<<endl;

	return 0;
}


/*
in:
5
0 1
0 2
1 3
1 4

out:
3
*/

【参考文献】
https://blog.csdn.net/hnjzsyjyj/article/details/119938620
https://blog.csdn.net/constantin_/article/details/79698270
https://blog.csdn.net/hnjzsyjyj/article/details/108655516