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

提示:

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

思路:返回二叉树中指定的区间的数之和,现将二叉树用中序遍历出来,然后将用循环语句判断是否在这个区间内,再相加即可。

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int rangeSumBST(TreeNode root, int L, int R) {
        List<Integer> al = new ArrayList<>();
        range(root,al);
        int sum = 0;
        for(int i = 0;i < al.size();i++){
            if(al.get(i) >= L && al.get(i) <= R){
                sum += al.get(i);
            }
        }
        return sum;
    }
    public void range(TreeNode root,List<Integer> list){        
        if(root != null){
            range(root.left,list);
            list.add(root.val);
            range(root.right,list);
        }
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_40904220/article/details/83957564
今日推荐