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

110.平衡二叉树

在这里插入图片描述

思路-递归:

  1. 明确递归函数的参数和返回值

参数:当前传入节点。 返回值:以当前传入节点为根节点的树的高度。

那么如何标记左右子树是否差值大于1呢?

如果当前传入节点为根节点的二叉树已经不是二叉平衡树了,还返回高度的话就没有意义了。

所以如果已经不是二叉平衡树了,可以返回-1 来标记已经不符合平衡树的规则了。

代码如下:

// -1 表示已经不是平衡二叉树了,否则返回值是以该节点为根节点树的高度
int getHeight(TreeNode* node)
  1. 明确终止条件

递归的过程中依然是遇到空节点了为终止,返回0,表示当前节点为根节点的树高度为0

代码如下:

if (node == NULL) {
    return 0;
}
  1. 明确单层递归的逻辑

如何判断以当前传入节点为根节点的二叉树是否是平衡二叉树呢?当然是其左子树高度和其右子树高度的差值。
分别求出其左右子树的高度,然后如果差值小于等于1,则返回当前二叉树的高度,否则返回-1,表示已经不是二叉平衡树了。

代码:用-1标记是否为平衡树,且当左右子树为-1或者差值>1时,都返回-1

///递归-后序遍历获得高度
// 递归
///递归-后序遍历获得高度
class Solution {
    public boolean isBalanced(TreeNode root) {
        return get(root)==-1?false:true;
    }
    int get(TreeNode root){
        // 
        if(root==null)return 0;
        int l=get(root.left);
        int r=get(root.right);
        // 注意:左右子树应该分别判断一下是不是平衡树
        if(Math.abs(l-r)>1||l==-1||r==-1){
            return -1;
        }else{
            return Math.max(l,r)+1;
        }
    }
}

思路二-迭代

模拟栈依次判断每个结点是否满足高度差,使用层序遍历获得高度

lass Solution {//统一迭代法
    public boolean isBalanced(TreeNode root) {
        Stack<TreeNode> stack=new Stack<>();
        if(root!=null){
            stack.push(root);
        }
        while(!stack.isEmpty()){
            TreeNode node = stack.peek();//中
            if(node!=null){
                node=stack.pop();
                stack.push(node);
                stack.push(null);
                if(node.right!=null){//右
                    stack.push(node.right);
                }
                if(node.left!=null){//左
                    stack.push(node.left);
                }
            }else{
                stack.pop();
                node=stack.pop();
                if(Math.abs(getheight(node.left)-getheight(node.right))>1){
                    return false;
                }
            }
        }
        return true;
    }
    // 层序遍历获得高度
    public int getheight(TreeNode root) {
        if (root == null) {
            return 0;
        }
        Deque<TreeNode> deque = new LinkedList<>();
        deque.offer(root);
        int depth = 0;
        while (!deque.isEmpty()) {
            int size = deque.size();
            depth++;
            for (int i = 0; i < size; i++) {
                TreeNode poll = deque.poll();
                if (poll.left != null) {
                    deque.offer(poll.left);
                }
                if (poll.right != null) {
                    deque.offer(poll.right);
                }
            }
        }
        return depth;
    }
}

257. 二叉树的所有路径

在这里插入图片描述

思路一:普通递归

path储存路径节点,res储存所有路径
因为每次递归会修改path数组list,所以递归时要进行回溯

class Solution {
    public List<String> binaryTreePaths(TreeNode root) {
        List<String> res = new ArrayList<>();
        if(root==null){
            return res;
        }
        List<Integer> path = new ArrayList<>();
        traversal(root,path,res);
        return res;
    }
    private void traversal(TreeNode node,List<Integer>paths,List<String> res){
        // 中,中为什么写在这里,因为最后一个节点也要加入到path中
        paths.add(node.val);
        // 找到叶子结点了,储存路径
        // 中
        if(node.left==null&&node.right==null){
            // StringBuilder用来拼接字符串,速度更快
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < paths.size() - 1; i++) {
                sb.append(paths.get(i)).append("->");
            }// 把path储存的数据逐一链接为路径
            // 记录做后一个节点
            sb.append(paths.get(paths.size()-1));
            // 收集一个路径
            res.add(sb.toString());
            return;
        }
        // 左
        if(node.left!=null){
        // 有递归就有回溯
            traversal(node.left,paths,res);// 递归
            paths.remove(paths.size() - 1);// 回溯
        }
        if(node.right!=null){
        // 有递归就有回溯
            traversal(node.right,paths,res);// 递归
            paths.remove(paths.size() - 1);// 回溯
        }
    }
}

思路二:递归优化

递归时把 tmp作为函数参数就是可以的,因为并没有改变String tmp的数值,执行完递归函数之后,tmp依然是之前的计算的数值(相当于回溯了)

class Solution {
    List<String> res=new ArrayList<>();
    public List<String> binaryTreePaths(TreeNode root) {
        deel(root,"");
        return res;
    }
    public void deal(TreeNode node,String s){
        if(node == null)return;
        if(node.left==null&&node.right==null){
            result.add(new StringBuilder(s).append(node.val).toString());
            return;
        
        }
        String tmp=new StringBuilder(s).append(node.val).append("->").toString();
        deal(node.left,tmp);
        deal(node.right,tmp);
    }
}

思路三:迭代法(没细看)

Java 可以直接定义一个成员变量为object的栈,这个栈可以放 TreeNode,可以放 String,这样就不需要搞两个栈,都放在一个栈里即可。
力扣题解+图解

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;
    }
}

404.左叶子之和

这道题目要求左叶子之和,其实是比较绕的,因为不能判断本节点是不是左叶子节点。

此时就要通过节点的父节点来判断其左孩子是不是左叶子了。

平时我们解二叉树的题目时,已经习惯了通过节点的左右孩子判断本节点的属性,而本题我们要通过节点的父节点判断本节点的属性。
在这里插入图片描述

思路-递归

判断当前节点是不是左叶子是无法判断的,必须要通过节点的父节点来判断其左孩子是不是左叶子。

如果该节点的左节点不为空,该节点的左节点的左节点为空,该节点的左节点的右节点为空,则找到了一个左叶子,判断代码如下:

if (node->left != NULL && node->left->left == NULL && node->left->right == NULL) {
    左叶子节点处理逻辑
}

递归的遍历顺序为后序遍历(左右中),是因为要通过递归函数的返回值来累加求取左叶子数值之和。

class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        if (root == null) return 0;
        int midValue = 0;
        // sumOfLeftLeaves(root.left)
        if(root.left!=null&&root.left.right==null&&root.left.left==null){
            midValue=root.left.val;
        }
        int leftValue = sumOfLeftLeaves(root.left);    // 左
        int rightValue = sumOfLeftLeaves(root.right);  // 右

        int sum = midValue + leftValue + rightValue;  // 中
        return sum;
    }
}

思路:统一迭代

在节点处判断逻辑

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
 // 递归
class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        // List<Integer> result = new LinkedList<>();
        Stack<TreeNode> st = new Stack<>();
        if (root != null) st.push(root);
        int res=0;
        while (!st.empty()) {
            TreeNode node = st.peek();
            if (node != null) {
                st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
                st.push(node);                          // 添加中节点
                st.push(null); // 中节点访问过,但是还没有处理,加入空节点做为标记。
                if (node.right!=null) st.push(node.right);  // 添加右节点(空节点不入栈)
                if (node.left!=null) st.push(node.left);    // 添加左节点(空节点不入栈)         
                               
            } else { // 只有遇到空节点的时候,才将下一个节点放进结果集
                st.pop();           // 将空节点弹出
                node = st.peek();    // 重新取出栈中元素
                st.pop();
                if(node.left!=null&&node.left.left==null&&node.left.right==null){
                    res+=node.left.val;
                }
                // result.add(node.val); // 加入到结果集
            }
        }
        return res;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值