代码随想录算法训练营day17|110 平衡二叉树| 257二叉树的所有路径| 404左叶子之和

110.平衡二叉树

力扣题目链接

给定一个二叉树,判断它是否是高度平衡的二叉树。

  • 平衡二叉搜索树:

    它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。

leetcode中强调的深度和高度很明显是按照节点来计算的,如图:

110.平衡二叉树2

关于根节点的深度究竟是1 还是 0,不同的地方有不一样的标准,leetcode的题目中都是以节点为一度,即根节点深度是1。但维基百科上定义用边为一度,即根节点的深度是0,我们暂时以leetcode为准(毕竟要在这上面刷题)。

思路:

要求比较高度,必然是要后序遍历。

三要素:

  1. 确定递归函数的参数和返回值: 参数:root 函数的返回类型为int 判断左右子树的高度 返回高度

  2. 确定终止条件: 当前节点为空的时候,就返回

  3. 确定单层递归的逻辑: 后序遍历,左右中

    分别求出其左右子树的高度,然后如果差值小于等于1,则返回当前二叉树的高度,否则返回-1,表示已经不是二叉平衡树了。

class Solution {
    public boolean isBalanced(TreeNode root) {
        return getHeight(root) != -1;
    }

    public int getHeight(TreeNode root) {
        //终止条件   空节点 高度是0
        if(root==null)return 0;
        
        //左
        int leftHeight=getHeight(root.left);
        if(leftHeight==-1) return -1;
        
        //右
        int rightHeight=getHeight(root.right);
        if(rightHeight==-1) return -1;
        
        // 左右子树高度差大于1,return -1表示已经不是平衡树了
        if(Math.abs(leftHeight-rightHeight)>1) return -1;
        return Math.max(leftHeight,rightHeight)+1;
    }
}

257. 二叉树的所有路径

力扣题目链接

给定一个二叉树,返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

思路:

根据题目要求:前序遍历 中左右

前序遍历以及回溯的过程如图:

257.二叉树的所有路径

我们先使用递归的方式,来做前序遍历。要知道递归和回溯就是一家的,本题也需要回溯。

三要素:

  • 确定递归函数的参数和返回值:

    传递根节点, 数组存单条路径, 数组存结果

  private void traversal(TreeNode root, List<Integer> paths, List<String> res) 
  • 确定终止条件:

    收集到叶子节点就结束了 左右孩子都为空就到叶子节点了

 if (root.left == null && root.right == null) {
            // 输出
            StringBuilder sb = new StringBuilder();// StringBuilder用来拼接字符串,速度更快
            for (int i = 0; i < paths.size() - 1; i++) {
                sb.append(paths.get(i)).append("->");
            }
            sb.append(paths.get(paths.size() - 1));// 记录最后一个节点
            res.add(sb.toString());// 收集一个路径
            return;
        }
  • 确定单层递归的逻辑: 中左右
paths.add(root.val);// 前序遍历,中        要写在判断终止条件前,如果写在后面会把叶子节点漏掉

// 递归和回溯是同时进行,所以要放在同一个花括号里
        if (root.left != null) { // 左
            traversal(root.left, paths, res);
            //递归完,做回溯,因为path 不能一直加入节点,它还要删节点,然后才能加入新的节点。
            paths.remove(paths.size() - 1);// 回溯
        }
        if (root.right != null) { // 右
            traversal(root.right, paths, res);
            paths.remove(paths.size() - 1);// 回溯
        }
        
class Solution {
    public List<String> binaryTreePaths(TreeNode root) {
        List<String> res=new ArrayList<>();  //放结果
        List<Integer> paths=new ArrayList<>();   //放路径
        if(root==null){
            return res;
        }
        traversal(root,paths,res);
        return res;

    }
    public void traversal(TreeNode node,List<Integer> paths,List<String> res){
        paths.add(node.val);
        if(node.left==null&&node.right==null){
            StringBuilder sb=new StringBuilder();
            for(int i=0;i<paths.size()-1;i++){
                sb.append(paths.get(i)).append("->");
            }
             sb.append(paths.get(paths.size()-1));
             res.add(sb.toString());
             return;
        }

        if(node.left!=null){
            traversal(node.left,paths,res);
            paths.remove(paths.size()-1);
        }
        if(node.right!=null){
            traversal(node.right,paths,res);
            paths.remove(paths.size()-1);
        }
    }
}

404.左叶子之和

力扣题目链接

计算给定二叉树的所有左叶子之和。

递归法

递归的遍历顺序为后序遍历(左右中),是因为要通过递归函数的返回值来累加求取左叶子数值之和。

递归三部曲:

  • 确定递归函数的参数和返回值

判断一个树的左叶子节点之和,那么一定要传入树的根节点,递归函数的返回值为数值之和,所以为int

  • 确定终止条件

如果遍历到空节点,那么左叶子值一定是0

 if (root == null) return 0;

注意,只有当前遍历的节点是父节点,才能判断其子节点是不是左叶子。 所以如果当前遍历的节点是叶子节点,那其左叶子也必定是0,那么终止条件为:

if (root.left != null && root.left.left == null && root.left.right == null) { 
            midValue = root.left.val;
  }
  • 确定单层递归的逻辑

当遇到左叶子节点的时候,记录数值,然后通过递归求取左子树左叶子之和,和 右子树左叶子之和,相加便是整个树的左叶子之和。

代码如下:

int leftValue = sumOfLeftLeaves(root.left);    // 左
//判断孩子是不是叶子节点
if (root.left != null && root.left.left == null && root.left.right == null) { 
      midValue = root.left.val;
}
int rightValue = sumOfLeftLeaves(root.right);  // 右
int sum =leftValue + rightValue;  // 中
 return sum;
class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        //终止条件
        if(root==null) return 0;
        //左
        int leftVaule=sumOfLeftLeaves(root.left);
        if(root.left!=null&&root.left.left==null&&root.left.right==null){
            leftVaule=root.left.val;
        }
        //右
        int rightVaule=sumOfLeftLeaves(root.right);
        //中
        int sum=leftVaule+rightVaule;
        return sum;

    }
}

//精简版代码
class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        //终止条件
        if(root==null) return 0;
        //左
        int leftVaule=0;
        if(root.left!=null&&root.left.left==null&&root.left.right==null){
            leftVaule=root.left.val;
        }
        return leftVaule+sumOfLeftLeaves(root.left)+sumOfLeftLeaves(root.right);

    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值