2017 小米笔试题 编程题 求树的高度 Java代码实现

首先声明 这篇文章是看到别的博主发表的博文 感觉这个题目挺有意思的 就用了另一种方式来实现 仅借用一下题目

附原文链接 :
点击打开链接

题目：树的高度

时间限制：C/C++语言 1000MS；其他语言 3000MS
内存限制：C/C++语言 65536KB；其他语言 589824KB
题目描述：
现在有一棵合法的二叉树，树的节点都是用数字表示，
现在给定这棵树上所有的父子关系，求这棵树的高度

输入
输入的第一行表示节点的个数n（1<=n<=1000，节点的编号为0到n-1）组成，
下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号

输出
输出树的高度，为一个整数

样例输入
5
0 1
0 2
1 3
1 4
样例输出
3

实现代码 :

嗯 ... 代码没有严谨的逻辑判断  因为懒

首先自定义一个二叉树类

public class Tree {

private int parentNum ;
private int myNum ;

public Tree() {
super();
// TODO Auto-generated constructor stub
}

public Tree(int parentNum, int myNum) {
super();
this.parentNum = parentNum;
this.myNum = myNum;
}

public int getParentNum() {
return parentNum;
}
public void setParentNum(int parentNum) {
this.parentNum = parentNum;
}
public int getMyNum() {
return myNum;
}
public void setMyNum(int myNum) {
this.myNum = myNum;
}
@Override
public String toString() {
return  parentNum + "  " + myNum ;
}

}

然后做初始化二叉树数据 和遍历 打印操作

public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int i = scanner.nextInt();

long l = System.currentTimeMillis();
//初始化二叉树
Tree[] trees = new Tree[i];

//parent为父节点
int parent = 0;
for (int j = 0; j < trees.length; j++) {

if (j == 0) {
//如果是顶层节点的二叉树
trees[j] = new Tree(parent, j);
} else {
trees[j] = new Tree(parent, j);
parent = j % 2 == 0 ? parent + 1 : parent;
}
}

for (int j = 0; j < trees.length; j++) {
System.out.println(trees[j]);
}

System.out.println(getLength(trees));
long m = System.currentTimeMillis();
System.out.println( m - l);

}

public static int getLength(Tree[] trees) {

//其实二叉树的层数 就是二叉树节点个数的二进制表示的长度
int nums = trees == null ? 0 : trees.length;

return Integer.toBinaryString(nums).length();
}

希望对大家有所帮助