代码随想录训练营day17

最新推荐文章于 2024-07-20 09:21:06 发布

JIUYU_M

最新推荐文章于 2024-07-20 09:21:06 发布

阅读量2.3k

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/weixin_45977348/article/details/127199682

版权

题目一：平衡二叉树

力扣题目链接

题目描述：给定一个二叉树，判断它是否是高度平衡的二叉树。

平衡二叉树定义为：一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1

思路分析：（具体步骤见代码随想录）

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

解法：递归


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

    private int getHeight(TreeNode root) {
        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;
    }
}

题目二：二叉树的所有路径

力扣题目链接

题目描述：给你一个二叉树的根节点 root ，按任意顺序，返回所有从根节点到叶子节点的路径

思路分析：（视频指路）

采用前序遍历，因为需要让父节点指向子节点，找到对应路径。题目中涉及到回溯，把路径记录下来，需要回溯来回退一一个路径在进入另一个路径。

解法一：递归


class Solution {
    public List<String> binaryTreePaths(TreeNode root) {
        List<String> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        List<Integer> paths = new ArrayList<>();
        traversal(root, paths, res);
        return res;
    }

    // paths:存放单条路径的数据  res:存放最终返回结果
    private void traversal(TreeNode root, List<Integer> paths, List<String> res) {
        // 为了收集到叶子节点，路径收集要放在终止条件上面
        paths.add(root.val);
        // 遍历到叶子节点，收获结果
        if (root.left == null && root.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 (root.left != null) {
            traversal(root.left, paths, res);
            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> result = new ArrayList<>();
        if (root == null)
            return result;
        Stack<Object> stack = new Stack<>();
        // 节点和路径同时入栈
        stack.push(root);
        stack.push(root.val + "");
        while (!stack.isEmpty()) {
            // 节点和路径同时出栈
            String path = (String) stack.pop();
            TreeNode node = (TreeNode) stack.pop();
            // 若找到叶子节点
            if (node.left == null && node.right == null) {
                result.add(path);
            }
            //右子节点不为空
            if (node.right != null) {
                stack.push(node.right);
                stack.push(path + "->" + node.right.val);
            }
            //左子节点不为空
            if (node.left != null) {
                stack.push(node.left);
                stack.push(path + "->" + node.left.val);
            }
        }
        return result;
    }
}

题目三：左叶子之和

力扣题目链接

题目描述：给定二叉树的根节点 root ，返回所有左叶子之和

左叶子定义：身为左孩子的叶子节点

思路分析：

找到左叶子的父节点，便可记录左叶子节点的数值

解法一：递归


class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        if (root == null) return 0;
        int leftValue = sumOfLeftLeaves(root.left);
        int rightValue = sumOfLeftLeaves(root.right);

        int midValue = 0;
        // 叶子节点为左孩子时，可记录其数值
        if (root.left != null && root.left.left == null && root.left.right == null) {
            midValue = root.left.val;
        }
        int sum = midValue + leftValue + rightValue;
        return sum;
    }
}

解法二：迭代


class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        if (root == null) return 0;
        Stack<TreeNode> stack = new Stack<>();
        stack.add(root);
        int res = 0;
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            if (node.left != null && node.left.left == null && node.left.right == null) {
                res += node.left.val;
            }
            if (node.right != null) stack.add(node.right);
            if (node.left != null) stack.add(node.left);
        }
        return res;
    }
}