平安科技的第一题不知道边界是什么情况。

死活都是百分之80,不知道是什么情况,有大佬可以帮看看么?
题目是给出父子关系,然后计算树的高度。感觉思路没问题。。。。。。

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

class TreeNode{
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val) {
this.val = val;
}
}


public class Main{


public static int getTreeHigh(TreeNode root) {
if(root == null)return 0;
return Math.max(getTreeHigh(root.left), getTreeHigh(root.right))+1;
}

public static void main(String[] args) {
Scanner input = new Scanner(System.in);
Map<Integer, TreeNode> mapTree = new HashMap<>();
int count = 0;
int rootVal = 0;
while(input.hasNextLine()) {
String guanxi = input.nextLine();
String[] fuAndZi = guanxi.split(" ");
int fatherNum = Integer.parseInt(fuAndZi[0]);
if(count == 0)rootVal = fatherNum;//根节点
int sun = Integer.parseInt(fuAndZi[1]);
TreeNode father = null;
if(mapTree.containsKey(fatherNum)) {
father = mapTree.get(fatherNum);
}else {
father = new TreeNode(fatherNum);
mapTree.put(fatherNum, father);
}
if(father.left != null) {
if(mapTree.get(sun) != null) {
father.right = mapTree.get(sun);
}else {
father.right = new TreeNode(sun);
mapTree.put(sun, father.right);
}
}else {
if(mapTree.get(sun) != null) {
father.left = mapTree.get(sun);
}else {
father.left = new TreeNode(sun);
mapTree.put(sun, father.left);
}
}
count++;
}
System.out.println(getTreeHigh(mapTree.get(rootVal)));
}
}


#平安科技#
全部评论
0 2 1 3 1 4 0 1这种情况
点赞 回复
分享
发布于 2018-09-16 22:38
根节点不一定是第一行那个父节点,考虑了吗
点赞 回复
分享
发布于 2018-09-16 23:02
滴滴
校招火热招聘中
官网直投
我的全a了 明天早上发
点赞 回复
分享
发布于 2018-09-17 01:32

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务