Leetcode938.二叉搜索树的范围和

1.题目

给定二叉搜索树的根结点 root,返回 LR(含)之间的所有结点的值的和。

二叉搜索树保证具有唯一的值。

示例 1:

输入:root = [10,5,15,3,7,null,18], L = 7, R = 15
输出:32

示例 2:

输入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出:23

2.解答

1。使用递归方式进行深入优先遍历

这道题相对比较简单,使用深入优先对树进行遍历,将其中符合范围的值进行累加即可,其中可以优化的地方是,加入当前节点的值小于L,那么这个节点的左子树不用再进行遍历了,因为左子树的所有节点一定是比根节点小的,直接遍历右子树即可,同理判断R,这是可以进行优化的一个地方

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private Integer sum;
    public int rangeSumBST(TreeNode root, int L, int R) {
        sum = 0;
        helper(root,L,R);
        return sum;
    }
    private void helper(TreeNode node,int L,int R){
        if(node == null)    return;       
        if(node.val >= L && node.val <= R)
            sum += node.val;
        if(node.val > L)
            helper(node.left,L,R);
        if(node.val < R)
            helper(node.right,L,R);
    }

}

2.使用迭代方式进行深入优先遍历

在解决树的问题时,不能见面一言不合就使用递归,可以多考虑一下迭代的方法,使用栈实现了树节点的深度优先搜索

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int rangeSumBST(TreeNode root, int L, int R) {
        //如果根节点为null,直接返回
        if(root == null)    return 0;
        //创建一个栈,并将根节点压入进行初始化
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        int sum = 0;
        while(!stack.isEmpty()){
            TreeNode temp = stack.pop();
            //弹出的节点符合范围,进行累加
            if(temp.val >= L && temp.val <= R)
                sum += temp.val;
            //进行优化,如果当前节点大于R,那么就不必对右子树进行遍历,因为右子树的所有节点一定大于根节点的
            if(temp.val < R && temp.right != null)
                stack.push(temp.right);
            //左子树同理
            if(temp.val > L && temp.left != null)
                stack.push(temp.left);           
        }
        return sum;
    }
}