树的高度
时间限制:1秒 空间限制:32768K
题目描述
现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度
输入描述:
输入的第一行表示节点的个数n(1 ≤ n ≤ 1000,节点的编号为0到n-1)组成, 下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号
输出描述:
输出树的高度,为一个整数
示例1
输入
5 0 1 0 2 1 3 1 4
输出
3
解题思路
直接dfs搜索,要保证只有两个子节点。具体见代码:
#include <stdio.h>
#include <string.h>
int s[1001][1001], ans[1001], max, n;
void dfs(int x, int step)
{
if (step > max)
max = step;
for (int y = x + 1; y < n; y++)
{
if (s[x][y])
{
ans[x]++;
if (ans[x] <= 2)
dfs(y, step + 1);
}
}
}
int main()
{
int x, y, min;
while (~scanf("%d", &n))
{
max = 1;
min = 99999999;
memset(s, 0, sizeof(s));
memset(ans, 0, sizeof(ans));
for (int i = 0; i < n - 1; i++)
{
scanf("%d %d", &x, &y);
s[x][y] = 1;
if (x < min)
min = x;
}
dfs(min, 1);
printf("%d\n", max);
}
return 0;
}