二叉树的深度
题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
输入
{1,2,3,4,5,#,6,#,#,7}
返回值
4
一看到题目其实我就想到要用递归了,但是一直想怎么也想不出来。真的,打代码最忌空想,其实你开始敲,思路就慢慢出来了。以后真就要改掉这个坏习惯。
方法一(递归,代码量很少,最重要的是解决思路)
/**
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) {
// 到了根节点就返回0
if(root == null)
return 0;
// 遍历左节点
int left = TreeDepth(root.left);
// 遍历右节点
int right = TreeDepth(root.right);
// 深度+1
return Math.max(left,right)+1;
}
}
方法二(非递归),利用队列,count是当前的节点,nextcount是当前深度总的节点。【总是要遍历到当前深度的最后一个节点,深度才加1】
import java.util.LinkedList;
import java.util.Queue;
public int TreeDepth1(TreeNode root) {
if(root==null) {
return 0;
}
Queue<TreeNode> q=new LinkedList<TreeNode>();
q.add(root);
int d=0,count=0,nextcount=q.size();
while(q.size()!=0) {
TreeNode t=q.poll();
count++;
if(t.left!=null) {
q.add(t.left);
}
if(t.right!=null) {
q.add(t.right);
}
// 当把当前深度的所有节点都遍历过后,深度加一,跳到下一深度
if(count==nextcount) {
d++;
// 当前节点重置
count=0;
nextcount=q.size(); // 保存最后节点
}
}
return d;
}