- 区间内查询数字的频率
请你设计一个数据结构,它能求出给定子数组内一个给定值的 频率 。
子数组中一个值的 频率 指的是这个子数组中这个值的出现次数。
请你实现 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);
rangeFreqQuery.query(0, 11, 33);
思路: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) {
if(!map.containsKey(value)) return 0;
List<Integer>list=map.get(value);
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;
l=0;r=list.size();
while(l<r){
int mid=(l+r)/2;
if(list.get(mid)<=right) l=mid+1;
else r=mid;
}
int right_pos=l;
return right_pos-left_pos;
}
}
|