求树的高度

2020-06-28  本文已影响0人  OakesYa

编码需求

现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度
输入描述:
输入的第一行表示节点的个数n(1 ≤ n ≤ 1000,节点的编号为0到n-1)组成,
下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号
输出描述:
输出树的高度,为一个整数
示例1
输入
5
0 1
0 2
1 3
1 4
输出
3

代码编程

import java.util.HashMap;
import java.util.Scanner;

public class TreeDepth {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        HashMap<Integer, Integer> deep = new HashMap<>();
        deep.put(0,1);
        int max = 1;
        int myDeep;
        for (int i=0;i< n-1;i++) {
            int parent = scanner.nextInt();
            int child = scanner.nextInt();
            if (!deep.containsKey(parent)) {
                continue;
            }
            myDeep = deep.get(parent) + 1;
            deep.put(child, myDeep);
            if (myDeep >max) {
                max = myDeep;
            }
        }
        System.out.println(max);
    }
}

统计到每个节点对应的高度即可,保留最大值即可

上一篇 下一篇

猜你喜欢

热点阅读