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

一句话总结：题解一看就懂，题目上手就废。

原题链接：110 平衡二叉树

原理是找到每个叶子节点的最大深度与最小深度只差是否大于1，那么需要递归地自顶向下地遍历每一层结点，找到其左右子树深度差。

class Solution {
    public boolean isBalanced(TreeNode root) {
        if (root == null) return true;
        else return Math.abs(height(root.left) - height(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right);
    }

    public int height(TreeNode root) {
        if (root == null) return 0;
        else return Math.max(height(root.left), height(root.right)) + 1;
    }
}

 原题链接：257 二叉树的所有路径

原理不难，用深度优先遍历可以很快地做出来。这里用的是广度优先遍历，利用两个队列存储结点和新的字符串，然后一边加入非叶子节点的子节点，一边将叶子节点的全部路径加入ans列表。

class Solution {
    public List<String> binaryTreePaths(TreeNode root) {
        List<String> ans = new ArrayList<>();
        Queue<TreeNode> node = new LinkedList<>();
        Queue<String> path = new LinkedList<>();
        node.offer(root);
        path.offer(Integer.toString(root.val));
        while (!node.isEmpty()) {
            TreeNode t = node.poll();
            String p = path.poll();
            if (t.left == null && t.right == null) ans.add(p);
            else {
                if (t.left != null) {
                    node.offer(t.left);
                    path.offer(new StringBuffer(p).append("->").append(t.left.val).toString());
                }
                if (t.right != null) {
                    node.offer(t.right);
                    path.offer(new StringBuffer(p).append("->").append(t.right.val).toString());
                }
            }
        }
        return ans;
    }
}

 原题链接：404 左叶子之和

无论是利用深度优先遍历还是广度优先遍历都可以很快得到答案。 这里用的是深度优先遍历。对左右子树分别递归地加入左叶子节点的值到ans中，返回即可。

class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        return root == null ? 0 : dfs(root);
    }

    int dfs(TreeNode root) {
        int ans = 0; 
        if (root.left != null) ans += isLeafNode(root.left) ? root.left.val : dfs(root.left);
        if (root.right != null && !isLeafNode(root.right)) ans += dfs(root.right);
        return ans;
    }

    boolean isLeafNode(TreeNode root) {
        if (root.left != null || root.right != null) return false;
        return true;
    }
}