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

题目描述

农场里有许多牛，每头牛都有一个观赏指数值，范围在1到10^4之间。农场主想知道，在给定的观赏指数区间内，哪些牛可以被安排到最佳观赏区。为了方便管理，农场主将牛的观赏指数按照二叉搜索树的形式排列。

给定一棵表示牛观赏指数的二叉搜索树的根节点 root 和一个整数区间 [low, high]，返回所有在区间内观赏指数的牛的观赏指数之和。

知识点

树，二叉搜索树，递归

解题思路

题目需要将在low和high范围内的节点值全部相加返回，树是二叉搜索树，左节点大于根节点大于右节点。

所以我们递归遍历整个树，如果当前节点的值小于low，就需要使用更大值的节点去判断了，所以传递右子树去接着递归。

如果当前节点值大于high，就需要使用更小节点去判断，所以传递左子树去接着递归。

而如果当前节点处于low和high之间，表示当前节点符合条件，ans相加当前节点值，左右子树都还有可能符合条件，左右子树也都拿去接着递归。 最后递归完之后的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 {
    public int rangeSumBST (TreeNode root, int low, int high) {
        // write code here
        int ans = 0;
        // root为null直接返回
        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;
        }
    }
}

总结

遇到树这种题，前中后序遍历一般需要使用递归来遍历全部结点，使用队列来层序遍历。 而此题遇到的是二叉搜索树，我们就需要知道二叉搜索树与普通树的性质差别，知道它左右节点值的关系后再通过比较与目标范围值的大小从而才能在遍历的过程中完成ans的获取。