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

目录

110 平衡二叉树

 递归

迭代

 257 二叉树的所有路径

 递归

迭代

404 左子叶之和 

递归

迭代

层序遍历迭代法

---

110 平衡二叉树

求深度:前序遍历

求高度:后序遍历

 递归

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(traversal(root) == -1)return false;
        return true;
    }
    //后序遍历求高度
    private int traversal(TreeNode root){
        if(root == null)return 0;
        int left = traversal(root.left);
        if(left == -1)return -1;
        int right = traversal(root.right);
        if(right == -1)return -1;
        if(Math.abs(left - right) > 1)return -1;
        return 1 + Math.max(left,right);//返回整个子树的最大值
    }
}

 时间复杂度O(n)

空间复杂度O(n)

迭代

class Solution {
     public boolean isBalanced(TreeNode root) {
        if (root == null) {
            return true;
        }
        Stack stack = new Stack<>();
        TreeNode pre = null;
        while (root!= null || !stack.isEmpty()) {
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
            TreeNode cur = stack.peek();
            // 右结点为null或已经遍历过
            if (cur.right == null || cur.right == pre) {
                // 比较左右子树的高度差，输出
                if (Math.abs(getHeight(cur.left) - getHeight(cur.right)) > 1) {
                    return false;
                }
                stack.pop();
                pre = cur;
                root = null;// 当前结点下，没有要遍历的结点了
            } else {
                root = cur.right;// 右结点还没遍历，遍历右结点
            }
        }
        return true;
    }
    public int getHeight(TreeNode root) {
        if(root == null)return 0;
        Dequest = new LinkedList<>();
        int depth = 0;
        st.offer(root);
        while(!st.isEmpty()){
            int siz = st.size();
            depth++;
            for(int i = 0;i < siz;i++){
                TreeNode cur = st.poll();
                if(cur.left != null)st.offer(cur.left);
                if(cur.right != null)st.offer(cur.right);
            }
        }
        return depth;
    }
}

 时间复杂度O(n^2)

空间复杂度O(n^2)

 257 二叉树的所有路径

 递归

class Solution {
    public List binaryTreePaths(TreeNode root) {
        Listres = new ArrayList<>();
        Listnums = new ArrayList<>();
        traversal(root,res,nums);
        return res;
    }
    private void traversal(TreeNode root,Listres,Listnums){
        //前序遍历 中左右
        nums.add(root.val);
        //达到子节点，将结果加入到res中
        if(root.left == null && root.right == null){
            StringBuilder sb = new StringBuilder();
            for(int i = 0;i < nums.size() - 1;i++){
                sb.append(nums.get(i)).append("->");
            }
            sb.append(nums.get(nums.size() - 1));
            res.add(sb.toString());
            return;
        }
        //左
        if(root.left != null){
            traversal(root.left,res,nums);//递归
            nums.remove(nums.size() - 1);//回溯
        }
        //右
        if(root.right != null){
            traversal(root.right,res,nums);
            nums.remove(nums.size() - 1);
        }
    }
}

时间复杂度O(n^2) 

空间复杂度O(n^2)

迭代

class Solution {
    public List binaryTreePaths(TreeNode root) {
        Listres = new ArrayList<>();
        if(root == null)return res;
        Deque