leetcode中等2080区间内查询数字的频率

Posted 2022-03-18 qq_40707462

2080. 区间内查询数字的频率
      请你设计一个数据结构，它能求出给定子数组内一个给定值的 频率 。

子数组中一个值的 频率 指的是这个子数组中这个值的出现次数。

请你实现 RangeFreqQuery 类：

RangeFreqQuery(int[] arr) 用下标从 0 开始的整数数组 arr 构造一个类的实例。
int query(int left, int right, int value) 返回子数组 arr[left…right] 中 value 的 频率 。
一个 子数组 指的是数组中一段连续的元素。arr[left…right] 指的是 nums 中包含下标 left 和 right 在内 的中间一段连续元素。

示例 1：

输入：

["RangeFreqQuery", "query", "query"]
[[[12, 33, 4, 56, 22, 2, 34, 33, 22, 12, 34, 56]], [1, 2, 4], [0, 11, 33]]
输出：
[null, 1, 2]

解释：
RangeFreqQuery rangeFreqQuery = new RangeFreqQuery([12, 33, 4, 56, 22, 2, 34, 33, 22, 12, 34, 56]);
rangeFreqQuery.query(1, 2, 4); // 返回 1 。4 在子数组 [33, 4] 中出现 1 次。
rangeFreqQuery.query(0, 11, 33); // 返回 2 。33 在整个子数组中出现 2 次

思路：hash记录每个元素出现的位置，在这个位置的list中，二分查找边界left和right的坐标，相减即可

注意，有点像34、在排序数组中查找元素的第一个和最后一个位置【中等】，但不太一样，34中要找的就是left和right的坐标，但本题中target（也就是left和right）不一定在list中出现

所以要找的是：

• 第一个大于等于left的位置
• 最大的<=right的位置

class RangeFreqQuery 
    Map<Integer,List<Integer>>map;
    public RangeFreqQuery(int[] arr) 
        map=new HashMap<>();
        for(int i=0;i<arr.length;i++)
            if(!map.containsKey(arr[i])) map.put(arr[i],new ArrayList<Integer>());
            map.get(arr[i]).add(i);//记录每个数字出现的位置
        
    
    
    public int query(int left, int right, int value) 
        //二分搜索list中left和right的位置
        if(!map.containsKey(value)) return 0;
        List<Integer>list=map.get(value);
        //第一个大于等于left的位置
        int l=0,r=list.size();
        while(l<r)
            int mid=(l+r)/2;
            if(list.get(mid)>=left) r=mid;
            else l=mid+1;
        
        int left_pos=l;
        
        //最大的<=right的位置
        l=0;r=list.size();
        while(l<r)
            int mid=(l+r)/2;
            if(list.get(mid)<=right) l=mid+1;//等于的话也加一，这样right_pos-left_pos就是个数，不用再加一了
            else r=mid;
        
        int right_pos=l;
        return right_pos-left_pos;