- 题目
树的高度
时间限制: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;
}