树的高度

时间限制: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;
}