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

最新推荐文章于 2024-11-04 19:24:37 发布

望仁啊

最新推荐文章于 2024-11-04 19:24:37 发布

阅读量133

点赞数

分类专栏：代码随想录算法刷题文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/m0_61863948/article/details/132752339

版权

代码随想录算法刷题专栏收录该内容

51 篇文章 0 订阅

订阅专栏

1 LeetCode 110 平衡二叉树

题目链接：LeetCode 110 平衡二叉树

文章讲解：代码随想录(programmercarl.com)

视频讲解：后序遍历求高度，高度判断是否平衡 | LeetCode：110.平衡二叉树

// 前序遍历顺序：中-左-右，入栈顺序：中-右-左
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null){
            return result;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()){
            TreeNode node = stack.pop();
            result.add(node.val);
            if (node.right != null){
                stack.push(node.right);
            }
            if (node.left != null){
                stack.push(node.left);
            }
        }
        return result;
    }
}

// 中序遍历顺序: 左-中-右 入栈顺序： 左-右
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null){
            return result;
        }
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;
        while (cur != null || !stack.isEmpty()){
           if (cur != null){
               stack.push(cur);
               cur = cur.left;
           }else{
               cur = stack.pop();
               result.add(cur.val);
               cur = cur.right;
           }
        }
        return result;
    }
}

// 后序遍历顺序 左-右-中 入栈顺序：中-左-右 出栈顺序：中-右-左， 最后翻转结果
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null){
            return result;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()){
            TreeNode node = stack.pop();
            result.add(node.val);
            if (node.left != null){
                stack.push(node.left);
            }
            if (node.right != null){
                stack.push(node.right);
            }
        }
        Collections.reverse(result);
        return result;
    }
}