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

题目描述

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

输入描述:

输入的第一行表示节点的个数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;
}

 

    原文作者:jirryzhang
    原文地址: https://blog.csdn.net/jirryzhang/article/details/82841353
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞