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

农场牛的最佳观赏区间

https://www.nowcoder.com/practice/7b49f5ad9814424d8c41de44f671d59e

考察知识点:二叉搜索树、二叉树遍历

解题思路:利用二叉搜索树左子树所有节点都比当前节点值小、右子树所有节点都比当前节点值大的性质,可以用于定位到符合条件的节点区间:即如果当前节点值比low小、或者当前节点值比high要大,都说明当前节点不在该区间内。这个时候如果比low小就应该增加当前节点值,也就是走右子树,比high大说明超了,要走左子树,最后再把所有符合条件的值相加即可。

另外也可以中序遍历 + 二分查找的方式来求得结果,借助上一题求得的顺序数组,这里就不做说明了。

本题解所用语言: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
        if (root == null) {
            return 0;
        }
        if (low > root.val) {
            return rangeSumBST(root.right, low, high);
        }
        if (high < root.val) {
            return rangeSumBST(root.left, low, high);
        }
        int leftSum = rangeSumBST(root.left, low, high);
        int rightSum = rangeSumBST(root.right, low, high);
        return root.val + leftSum + rightSum;
    }
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务