多叉树的构建和树的高度的计算

题目描述

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

输入描述:

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

输出描述:

输出树的高度，为一个整数

示例1

输入

5
0 1
0 2
1 3
1 4

输出

3

/*
题目有点小问题，测试用例存在多叉树但按照 AC 代码的意思要剔除这样的例子(-_-||
但我总结了一下对于这种输入如何构建一个多叉树，以及找出最大高度的方法
可能方法不是最好的，希望各位批评指正！！ 
对于题目中的输入
0 1
0 2
1 3
1 4
左边是父节点，右边是子节点
一个父亲可能有多个儿子，我这里用的是 map<int,queue<int> >来对应这种关系
以父亲节点的值为 key, 所对应的一个队列queue<int>里放着他所有的儿子 
这样就建立好了基本的对应关系，找出最大高度的方法类似于广度优先遍历
一层一层向下迭代，详见代码 
名词解释： 
一个节点的高度：从该节点到一个树叶(没有儿子的节点)的最长距离 
*/
#include<iostream>
#include<map>
#include<queue>
#include<vector>
#include<algorithm> 
#include<string.h>
using namespace std;
map<int,queue<int> > G;//存储节点对应关系的 map 
bool cmp(int a,int b)//自定义比较函数,对应sort的降序排列 
{
	return a>b;
}
/*
获取最大高度的方法 
*/
int getMax(int begin)		//函数传入的是根节点 
{
	if(G[begin].empty())
	return 1; 		//这是递归结束条件，即遍历到树叶(没有儿子的节点)返回高度 1 
	vector<int> v;		//一个节点有多个儿子，这是一个记录多个儿子高度的动态数组 
	while(!G[begin].empty())		//把每一个儿子的高度放进vector 
	{
		int length = getMax(G[begin].front()) + 1;		//往下一层，高度 + 1 
		v.push_back(length);		//儿子节点的高度放入vector 
		G[begin].pop();		//注意pop()遍历过的节点 
	}
	//求动态数组的最值 ,作为该节点的最大高度并返回
	sort(v.begin(),v.end(),cmp);//降序排序
	return v[0]; //第一个即为最大值 
} 
int main()
{
	freopen("input.txt","r",stdin);
	int a,b,n,begin;
	int num[1001]={0};
	while(~scanf("%d",&n))
	{
		//多叉图的构建 
		for(int i=0;i<n-1;i++)
		{
			scanf("%d %d",&a,&b);
			G[a].push(b);
			num[b] = 1;		//散列思想，把所以儿子节点在数组 num里置 1,这样没被置 1的就是根节点 
		}
		//根节点的获取
		//遍历 num数组，不是 1的就是根节点 
		for(int i=0;i<n;i++)
		{
			if(num[i]==0)
			{
				begin = i;
				break;
			}
		} 
		//遍历树的最大高度
		printf("%d\n",getMax(begin)); //放入根节点 
		memset(num,0,sizeof(num));	//多组输入，初始化全局变量 
		 
	}

	return 0;
}