题解 | #农场牛的最佳观赏区间#

农场牛的最佳观赏区间

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;
        }
    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务