无尽算法之 二叉搜索树的范围和

给定二叉搜索树的根结点 root,返回 L 和 R(含)之间的所有结点的值的和。
二叉搜索树保证具有唯一的值。

示例 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

提示:

树中的结点数量最多为 10000 个。
最终的答案保证小于 2^31。

思路:

DFS大法好
遍历所有节点, 若L<=node.val<=R, 计入和中, 否则丢掉值.

题解:

解法一 直观解法

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

解法二 充分利用递归机制

class Solution {
    int sum=0;
    public int rangeSumBST(TreeNode root, int L, int R) {
        if(root==null){
            return 0;
        }
        int temp=0;
        if(root.val>=L&&root.val<=R){
            temp=root.val;
        }
        return rangeSumBST(root.left,L,R)+rangeSumBST(root.right,L,R)+ temp;
    }
}
发布了126 篇原创文章 · 获赞 239 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq_33709508/article/details/103934909