题解 | #农场牛的最佳观赏区间#
农场牛的最佳观赏区间
https://www.nowcoder.com/practice/7b49f5ad9814424d8c41de44f671d59e
知识点
树,二叉搜索树
解题思路
题目需要将在low和high范围内的节点值全部相加返回,树是二叉搜索树,左节点大于根节点大于右节点。
所以我们层序遍历整个树,如果当前节点的值小于low,就需要使用更大值的节点去判断了,所以传递右子树去接着递归。
如果当前节点值大于high,就需要使用更小节点去判断,所以传递左子树去接着递归。
而如果当前节点处于low和high之间,表示当前节点符合条件,ans相加当前节点值,左右子树都还有可能符合条件,左右子树也都拿去接着递归。
Java题解
import java.util.*; /* * public class TreeNode { * int val = 0; * TreeNode left = null; * TreeNode right = null; * public TreeNode(int val) { * this.val = val; * } * } */ public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root TreeNode类 * @param low int整型 * @param high int整型 * @return int整型 */ public int rangeSumBST (TreeNode root, int low, int high) { // write code here int ans = 0; if(root == null) return ans; if(root.val < low){ return rangeSumBST(root.right,low,high); } else if(root.val > high){ return rangeSumBST(root.left,low,high); } else { ans += root.val; ans += rangeSumBST(root.left,low,high); ans += rangeSumBST(root.right,low,high); return ans; } } }