代码随想录算法训练营第十七天｜110.平衡二叉树、257. 二叉树的所有路径、404.左叶子之和

代码随想录算法训练营第十七天｜110.平衡二叉树、257. 二叉树的所有路径、404.左叶子之和

110.平衡二叉树

题目链接

问题简述：判断二叉树是否为平衡二叉树。

思考：判读二叉树遍历方式为左右根，求高度遍历方式也为左右根。

算法思路：如果当前节点为空或者为叶子结点，则返回true；如果不为空或者为叶子结点，则判断左右子树高度差十分大于1，大于一则直接返回fasle，小于一要判断左右子树收否都为平衡二叉树，如果不是则返回fasle。

class Solution {
    public boolean isBalanced(TreeNode root) {
        if (root == null) return true;
     		//叶子结点为平衡二叉树
        if (root.left == null && root.right == null) return true;
      	//左右子树高度大于1不为平衡二叉树
        if (Math.abs(getdeep(root.left) - getdeep(root.right)) > 1) return false;
      	//当前节点的左右子树高度小于1，且左右子树均为为平衡二叉树，返回true
        if (isBalanced(root.right) && isBalanced(root.left)) return true;
      	//左右子树不都为平衡二叉树，返回fasle
        return false;
    }
  	//求二叉树当前节点高度
    public int getdeep(TreeNode root){
        if (root == null) return 0;
        return Math.max(getdeep(root.left), getdeep(root.right)) + 1;
    }
}

257. 二叉树的所有路径

题目链接

问题简述：得到二叉树的所有路径。

思考：这道题感觉很难理解，花了不少时间。关键在于不理解回溯。然后路径都是根结点到叶子结点的路径，所以要先遍历跟结点，再遍历左右结点。

算法思路：先序遍历每一个结点，将结点放入path，如果遍历到叶子结点时，将路径path存入StringBuilder，再存入res。如果不是叶子结点，则遍历左结点，递归添加左面结点，存入所有路径后进行回溯。再递归添加右面结点，存入所有路径后进行回溯。最后返回。

import java.util.ArrayList;
import java.util.List;

class Solution {
    public List<String> binaryTreePaths(TreeNode root) {
        //用于存储最终结果
        List<String> res = new ArrayList<>();
        //用于存储每条路径值的List
        List<Integer> path = new ArrayList<>();
        traversal(root, path, res);
        return res;
    }

    public void traversal(TreeNode root, List<Integer> path, List<String> res){
        if (root == null) return;
      	//path中加入当前节点
        path.add(root.val);
      	//到达叶子结点
        if (root.left == null && root.right == null){
            //此时已到达叶子结点，将path的元素传入StringBuilder
            StringBuilder sb = new StringBuilder();
            //先添加一个元素
            sb.append(path.get(0));
            //再不断添加间隔和数字
            for (int i = 1; i < path.size(); i++) {
                sb.append("->").append(path.get(i));
            }
            //将字符串加入List
            res.add(sb.toString());
            return;
        }
      	//左面有结点先添加左面结点
        if (root.left != null){
          	//递归添加左面结点添加
            traversal(root.left, path, res);
          	//回溯到添加这个左结点之前（开始一直没想明白为什么回溯只需要删除一个结点，而不是删除当前root后的所有结点：是因为每次递归都会删除当前添加的一个结点，所以这时候path就只剩下多余的一个结点了）
            path.remove(path.size() - 1);
        }
        if (root.right != null){
          	//递归添加左面结点
            traversal(root.right, path, res);
          	//回溯到添加这个右结点之前
            path.remove(path.size() - 1);
        }
      	//遍历完了所有路径
        return;
    }
}

404.左叶子之和

题目链接

问题简述：求出所有左叶子结点和。

思考：这道题也是开始少考虑了，看了leetcode的错误用例才明白错在哪。

算法思路：定义一个sum，当遍历空结点时直接返回0；当前结点为叶子结点时也直接返回0；当遍历到左孩子是叶子结点的的结点时，sum添加当前值。再一次遍历左右子树，将所有左叶子找出添加进sum。

根左右遍历：

class Solution {
    int sum = 0;
    public int sumOfLeftLeaves(TreeNode root) {
        if (root == null) return 0;
      	//当前结点为叶子结点时也直接返回0，因为在遍历这个结点的父结点时就已经将其判断添加sum
        if (root.left == null && root.right == null) return 0;
      	//当遍历到左孩子是叶子结点的的结点时，sum添加当前值
        if (root.left != null && root.left.left == null && root.left.right == null) sum += root.left.val;
      	//遍历左右结点
        sumOfLeftLeaves(root.left);
        sumOfLeftLeaves(root.right);
        return sum;
    }
}

算法思路：当遍历空结点时直接返回0；当前结点为叶子结点时也直接返回0；当遍历到左孩子是叶子结点的的结点时，返回左孩子值与右子树值加合。如果左子树不是叶子结点，则返回两个树的加合

后序遍历：

class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        if (root == null) return 0;
      	当前结点为叶子结点时也直接返回0
        if (root.left == null && root.right == null) return 0;
      	//当遍历到左孩子是叶子结点的的结点时，直接返回左孩子值与右子树的加合，且此时要保证左子树存在
        if (root.left != null && root.left.left == null && root.left.right == null) return root.left.val + sumOfLeftLeaves(root.right);
      	//如果左子树不是叶子结点，则返回两个树的加合
        return sumOfLeftLeaves(root.left) + sumOfLeftLeaves(root.right);
    }
}

感想

感觉第二题有难度，可能自己想不出来，边看代码示例边写的。