今天开始学Java 现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度

题目描述

现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度

输入描述:

输入的第一行表示节点的个数n(1 ≤ n ≤ 1000,节点的编号为0到n-1)组成,
下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号

输出描述:

输出树的高度,为一个整数
示例1

输入

5
0 1
0 2
1 3
1 4

输出

3

这道题开始我是想到用tree型结构来做,但是代码写到后来编不下去了,感觉这条路行不通(因为题目中需要查找父节点)。于是看了其他人的思路,发现有人用HashMap来做,构造两个hashmap分别来存放父节点和其子节点的个数,子节点以及子节点的深度。其实就是用了类似于数组索引的思想。//做题时,不要被其表面所迷惑,要先自己想想还有没有可能用更方便的方法

 public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        while(sc.hasNext()){
            int num = sc.nextInt();
            HashMap<Integer,Integer> tree = new HashMap();
            HashMap<Integer,Integer> child_deep = new HashMap();
            //树一开始根结点为0,孩子结点个数为0
            tree.put(0,0);
            //孩子结点为0,深度为1
            child_deep.put(0,1);
            int dep = 0;
            int max = 1;
            for (int i = 0;i<num-1;i++){
                int father = sc.nextInt();
                int son = sc.nextInt();
                //ArrayList里面是contains
                //树里面不包含父节点或者儿子已满的情况,忽略
                if(!tree.containsKey(father) || tree.get(father) >=2  ){
                    continue;
                }
                    //父亲的孩子数+1
                    tree.put(father,tree.get(father)+1);
                    //孩子变为另一个父亲
                    tree.put(son,0);
                    //深度+1
                    child_deep.put(son,child_deep.get(father)+1);
                    dep = child_deep.get(son);
                
            if(max<dep ){
                    max = dep;
                }
            }
            System.out.println(max);
            }
            
        }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值