题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
题解:
递归解法:
后序遍历二叉树
public class Solution {
public int TreeDepth(TreeNode root) {
if(root==null){
return 0;
}
int left=TreeDepth(root.left);
int right=TreeDepth(root.right);
return 1+(left>right?left:right);
}
}
非递归解法:
层次遍历
在原有使用队列层次遍历二叉树的基础上,增加两个变量
lastNode 指向当前行的最右节点
nlastNode 指向下一行的最右节点
每当输出一个节点后,将把此节点的左右孩子压入队列,压入队列时,让 nlastNode 跟踪入队的节点,然后判断 刚输出的节点是否是当前行的最右节点(node==lastNode?),如果是,此时 nlastNode 指向的节点一定是下一行的最右节点,表示一行遍历结束,depth++,将 nlastNode 赋给 lastNode。继续出栈,输出下一行。
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<TreeNode>();
queue.add(root);
int depth=0;
TreeNode lastNode=root;//指向当前行的最右节点
TreeNode nlastNode=root;//指向下一行的最右节点
while(!queue.isEmpty()){
TreeNode node=queue.poll();
if(node.left!=null){
nlastNode=node.left;
queue.add(node.left);
}
if(node.right!=null){
nlastNode=node.right;
queue.add(node.right);
}
if(node==lastNode){
depth++;
lastNode=nlastNode;
}
}
return depth;
}
}