二叉搜索树的范围和(深度优先遍历)

二叉搜索树的范围和

题目描述

给定二叉搜索树的根结点 root，返回值位于范围 [low, high] 之间的所有结点的值的和。

示例 1：

输入：root = [10,5,15,3,7,null,18], low = 7, high = 15
输出：32

题解（java）【深度优先遍历】

class Solution {
    int res = 0;
    public int rangeSumBST(TreeNode root, int L, int R) {
        if (root != null){
            if(root.val < L){//当前小于L，则只需要遍历右子树即可
                rangeSumBST(root.right, L, R);
            }else if(root.val > R){//当前大于R，则只需要遍历左子树即可
                rangeSumBST(root.left, L, R);
            }else{//当前在L~R，则两个子树都要遍历
                res += root.val;
                rangeSumBST(root.left, L, R);
                rangeSumBST(root.right, L, R);
            }
        }
        return res;
    }
}

图示

详解

• 1>.首先，我们需要定义一个变量 rec 来记录符合范围节点的和
• 2>.题目给定范围中的最小值为 L,最大值为 R,当 当前遍历的该节点值小于 L 时，那么该节点的左子树全都小于 L,我们继续遍历该节点的右子树即可，且该节点的值不计入总和中。
• 3>.当 当前遍历的节点值大于 R 时，那么该节点的右子树全部大于 R,我们继续遍历该节点的左子树即可，且该节点的值不计入总和中。
• 4>.当 当前遍历的节点值位于 L 与 R 中间时，先把该节点的值计入总和中，然后在分别遍历左子树和右子树。
• 5>.当遍历到 root == null 时，直接返回 rec 即可

声明

• 原作者：E.L.E
• <本文章著作权归作者所有，商业转载请获得作者授权，非商业转载请注明出处>
• <欢迎大家评论>