输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } }
注意:树的深度,是要求最长的树高,即要找到所有的叶子结点
java中的选择两数中的最大值,要用Math.max(a, b)
public class Solution { public int TreeDepth(TreeNode root) { return getDeep(root, 0); } public int getDeep(TreeNode root, int len){ if(root == null) return 0; if(root.left == null && root.right == null){ return len + 1; } return 1 + Math.max(getDeep(root.left, len), getDeep(root.right, len)); } }
看了别人的代码后,改进自己的方法
public class Solution { public int TreeDepth(TreeNode root) { if(root == null) return 0; return 1 + Math.max(TreeDepth(root.left), TreeDepth(root.right)); } }
别人的非递归方法
采用层序遍历,每读取一层,level++.
同时采取队列保存上一层的所有节点
队列要用实现了队列接口的类来完成,如LinkedList 等
剑指offer——栈与队列相关 Queue,Deque,Stack
import java.util.Queue; import java.util.LinkedList; public class Solution { public int TreeDepth(TreeNode root) { if(root == null) return 0; Queue<TreeNode> queue = new LinkedList<>(); queue.offer(root); int level = 0; while(queue.peek() != null){ int preCount = queue.size(); while((preCount--) != 0){ TreeNode node = queue.poll(); if(node.left != null){ queue.offer(node.left); } if(node.right != null){ queue.offer(node.right); } } level++; } return level; } }