小米 秋招 笔试 编程题 树的高度

• 题目
  树的高度
  时间限制：1秒 空间限制：32768K 热度指数：34988
  算法知识视频讲解
  校招时部分企业笔试将禁止编程题跳出页面，为提前适应，练习时请使用在线自测，而非本地IDE。
  题目描述
  现在有一棵合法的二叉树，树的节点都是用数字表示，现在给定这棵树上所有的父子关系，求这棵树的高度
  输入描述:
  输入的第一行表示节点的个数n（1 ≤ n ≤ 1000，节点的编号为0到n-1）组成，
  下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号
  输出描述:
  输出树的高度，为一个整数
  示例1
  输入
  复制
  5
  0 1
  0 2
  1 3
  1 4
  输出
  复制
  3
• 解法
  代码

#include<iostream>
#include<map>

using namespace std;

int height_tree(int root,int** relation)
{
    int l_h,r_h;
    if(root==-1)
        return 0;
    if(relation[root][0]==-1&&relation[root][1]==-1)
        return 1;
    
    l_h=height_tree(relation[root][0],relation);
    r_h=height_tree(relation[root][1],relation);

    return l_h>r_h?l_h+1:r_h+1;
}


int main()
{
    int n,i,father,son;
    cin>>n;

    int **relation=new int*[n];
    for(i=0;i<n;i++)
        relation[i]=new int[2];
    for(i=0;i<n;i++)
    {
        relation[i][0]=-1;
        relation[i][1]=-1;
    }
    
    i=1;
    while(i<n)
    {
        cin>>father>>son;
        if(relation[father][0]==-1)
            relation[father][0]=son;
        else 
        {
            if(relation[father][1]==-1)
                relation[father][1]=son;
        }
        i++;
    }

    cout<<height_tree(0,relation)<<endl;
    
    return 0;
}