Problem : https://leetcode.com/problems/range-sum-of-bst/
Use BST property to trim branches outside of the range.
/**
* 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 rangeSumBST(TreeNode root, int low, int high) {
if (root == null) return 0;
// left subtree and current node are out of the range
if (root.val < low) return rangeSumBST(root.right, low, high);
// right subtree and current node are out of the range
if (root.val > high) return rangeSumBST(root.left, low, high);
// current node is inside of the range
return rangeSumBST(root.left, low, high) + root.val + rangeSumBST(root.right, low, high);
}
}
No comments:
Post a Comment