题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
思路:
刚刚开始我使用的是一个非递归的算法,我想着这个是一个求深度的二叉树,那么我就去中序遍历这个二叉树,然后使用一个一个栈进行消除递归,在遍历的过程中只要栈中进一个元素那么辅助统计 i 就加一,此时定义一个统计maxDep始终和这个i进行比较,当i > maxDep; 此时将 i 赋给 maxDep。
之后在牛客网上看到一个比较好算法,是一个递归的算法,这个算法利用的是一个递归的思想,遍历一个节点,加一的基础上求左子树和右子树高度的最大值。
非递归实现:
import java.util.Stack;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
/*使用非递归的方式*/
public int TreeDepth(TreeNode root) {
Stack<TreeNode> stack = new Stack<TreeNode>();
if(root == null){
return 0;
}
TreeNode p = root;
int i = 0;
int maxDep = 0;
while(p != null || !stack.isEmpty()){
if(p != null){
stack.push(p);
i ++;
if(i > maxDep){
maxDep = i;
}
p = p.left;
}else{
p = stack.pop();
if(p.right == null){
i --;
}
p = p.right;
}
}
return maxDep;
}
递归实现:
import java.util.Stack;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
/*使用非递归的方式*/
/*这种是使用递归的方式*/
public int TreeDepth(TreeNode root) {
/*使用非递归的方式*/
if(root == null)
return 0;
return Math.max(1 + TreeDepth(root.left), 1 + TreeDepth(root.right));
}
}