题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
解法1:递归版,首先,给出递归结束的条件,当树为空时,深度为零,接下来,就是一层一层的递归了,比较左右子数的深度,其中值较大的作为当前子节点所在层的深度,则父节点的深度为子节点的深度 + 1;
public class Solution {
public int TreeDepth(TreeNode root) {
if (root == null)
return 0;
int left = TreeDepth(root.left);
int right = TreeDepth(root.right);
return left > right ? left + 1 : right + 1;
}
}
解法二:非递归版,使用双辅助栈,一个用来存储节点,一个用来存储当前节点的深度,同时,定义一个变量用来存储当前所找到的最大深度,然后,在节点出栈的同时,将其当前深度也出栈,比较当前深度与最大深度的值,更新最大深度值,然后,将当前节点的左右节点及其深度也分别入栈,最后,直到栈为空时,得到的最大深度就是该树的深度。
public class Solution {
public int TreeDepth(TreeNode root) {
if (root == null)
return 0;
Stack<Integer> depth = new Stack<Integer>();
Stack<TreeNode> node = new Stack<TreeNode>();
int curDepth = 0;//当前节点深度
int maxDepth = 0;//当前最大深度
node.push(root);
depth.push(1);
while (!node.isEmpty()){
TreeNode curNode = node.pop();
curDepth = depth.pop();
if (curDepth > maxDepth)
maxDepth = curDepth;
if (curNode.left != null){
node.push(curNode.left);
depth.push(curDepth + 1);
}
if (curNode.right != null){
node.push(curNode.right);
depth.push(curDepth + 1);
}
}
return maxDepth;
}
}