训练营练习-Day17&二叉树4

Leetcode 110 力扣

解法：递归三部曲

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

private int getHeight(TreeNode root) {
    if (root == null) {
        return 0;
    }
    int leftNum = getHeight(root.left);
    if (leftNum == -1) {  // 左子树为-1表示已经不是平衡树了
        return -1;
    }
    int rightNum = getHeight(root.right);
    if (rightNum == -1) { // 右子树为-1表示已经不是平衡树了
        return -1;
    }
    // 左右子树高度差大于1，return -1表示已经不是平衡树了
    if (Math.abs(leftNum - rightNum) > 1) {
        return -1;
    }
    return Math.max(leftNum, rightNum) + 1;
}

Leetcode 257 力扣

题目需要视频理解递归中带着回溯，你感受到了没？| LeetCode：257. 二叉树的所有路径_哔哩哔哩_bilibili

解法一：返回树的所有路径中左右，用前序遍历，递归+回溯

public List binaryTreePaths(TreeNode root) {

        List res = new ArrayList<>();

        if (root == null){

            return res;

        }

        List path = new ArrayList<>();

        traversal(root,path,res);

        return res;

    }

    private void traversal(TreeNode root, List path, List res) {

        path.add(root.val); // 中

        if (root.left == null && root.right == null){ // 叶子节点收集结果

            StringBuilder sb = new StringBuilder();

            for (int i = 0; i < path.size()-1; i++) {

                sb.append(path.get(i)).append("->");

            }

            sb.append(path.get(path.size()-1)); // 收集最后一个节点，不需要->拼接

            res.add(sb.toString());

            return;

        }

        //path.add(root.val); // 中不能放在递归终止条件之后，会遗漏当前叶子节点val

        if(root.left != null){ // 左

            traversal(root.left,path,res); // 递归

            path.remove(path.size()-1); // 回溯

        }

        if(root.right != null){ // 右

            traversal(root.right,path,res); // 递归

            path.remove(path.size()-1); // 回溯

        }

    }

解法二：迭代使用栈

/**
 * 迭代法
 */
public List binaryTreePaths1(TreeNode root) {
    List result = new ArrayList<>();
    if (root == null)
        return result;
    Stack