题目一:二叉树的深度
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
思路:嘿嘿,递归简简单单几行搞定,有点类似展开树求最大值。
/**
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(TreeDepth(root.left), TreeDepth(root.right)) + 1;
}
}
- 时间复杂度:O(n)。
- 空间复杂度:O(logn)。
练习地址:https://www.nowcoder.com/practice/435fb86331474282a3499955f0a41e8b
题目二:平衡二叉树
输入一棵二叉树,判断该二叉树是否是平衡二叉树。如果某二叉树中任意节点的左、右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
是的,还是递归,跟上面题目一不同的是我们需要优化一下,即遍历到左右深度相差大于1时就停止递归了。较少复杂度
public class Solution {
int totalCount=0; // 使用变量记录当前比较的两个分支的深度值
public boolean IsBalanced_Solution(TreeNode root) {
if(root==null){
totalCount = 0;
return true;
}
// 当前短枝已经不满足条件则不再递归
if(!IsBalanced_Solution(root.left)){
return false;
}
int left = totalCount;
// 当前短枝已经不满足条件则不再递归
if(!IsBalanced_Solution(root.right)){
return false;
}
int right = totalCount;
int diff = left - right;
if(diff<=1&&diff>=-1){
totalCount = 1+(left>right?left:right); // 赋值
return true;
}else{
return false;
}
}
}
- 时间复杂度:O(n)。
- 空间复杂度:O(logn)。