由树节点的父子关系计算树的深度(据说是小米笔试题)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/jirryzhang/article/details/82841353

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

5
0 1
0 2
1 3
1 4

输出

3

/*父节点的深度加一就是子节点的深度，从0开始一次往下遍历*/
/*0的深度为1，一次以此为基础向下遍历，则可得到最终深度 */ 
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
 
int main()
{
	int n;
	while(cin>>n){
        int height=0;//记录全局最大深度
        int result[1000]={1};//这两行数组大小固定为1000是因为用例的节点编号并不是连续的，0`999的编号都有可能出现
        int cnt[1000]={0};
        vector<vector<int>> v(n-1,vector<int>(2));
        for(int i=0;i<n-1;i++){
            cin>>v[i][0];
            cin>>v[i][1];
        }
        sort(v.begin(),v.end());

        for(int i=0;i<n-1;i++){
            if(cnt[v[i][0]]<2){//由于测试用例要求每个父节点只有两个子节点，因此两个以外的子节点直接舍弃
                result[v[i][1]]=result[v[i][0]]+1;//根据父节点深度计算子节点所在深度
                if(result[v[i][1]]>height)
                    height=result[v[i][1]];
                ++cnt[v[i][0]];//根据父节点的子节点个数
            }
        }
        cout<<height<<endl;
    }
    return 0;
}