LeetCode.938. 二叉搜索树的范围和

给定二叉搜索树的根结点 root,返回 L 和 R(含)之间的所有结点的值的和。

二叉搜索树保证具有唯一的值。

示例 1:

输入:root = [10,5,15,3,7,null,18], L = 7, R = 15
输出:32
示例 2:

输入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出:23
 

提示:

树中的结点数量最多为 10000 个。
最终的答案保证小于 2^31。

public class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode(int x) { val = x; }
      public TreeNode getLeft(){
          return this.left;
      }
      public TreeNode getRight(){
          return this.right;
      }
      public int getVal(){
          return this.val;
      }
}

class Solution {
    List<Integer> rootVals=new ArrayList<>();
    int result=0;

    public int rangeSumBST(TreeNode root, int L, int R) {
       helper(root);

       for(int i=0;i<rootVals.size();i++){
           if(rootVals.get(i).intValue()>=L && rootVals.get(i).intValue()<=R){
             result=result+rootVals.get(i);
           }
       }
       return result;
    }

    public void helper(TreeNode root){
      if(root!=null){
          helper(root.getLeft());
          rootVals.add(root.getVal());
          helper(root.getRight());
      }
    }
}

猜你喜欢

转载自blog.csdn.net/lzc2644481789/article/details/108760515