题目
请你设计一个数据结构,它能求出给定子数组内一个给定值的 频率 。
子数组中一个值的 频率 指的是这个子数组中这个值的出现次数。
请你实现 RangeFreqQuery 类:
RangeFreqQuery(int[] arr) 用下标从 0 开始的整数数组 arr 构造一个类的实例。
int query(int left, int right, int value) 返回子数组 arr[left…right] 中 value 的 频率 。
一个 子数组 指的是数组中一段连续的元素。arr[left…right] 指的是 nums 中包含下标 left 和 right 在内 的中间一段连续元素。
示例
输入:
[“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 次。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-frequency-queries
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:哈希表+二分查找
1、将所有 值 和 相同值对应的索引 存进一个哈希表里。
2、对索引进行二分查找。
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<>());
map.get(arr[i]).add(i);
}
}
public int query(int left, int right, int value) {
//要查找的值不在数组里,返回0
if (!map.containsKey(value)) return 0;
//要找的值的所有索引下标的集合(天然有序)
List<Integer> index = map.get(value);
int n = index.size();
//找第一个index[i]大于等于left的i
int l = 0, r = n - 1;
int flagl = -1;
while (l < r) {
int mid = l + r >> 1;
if (index.get(mid) >= left) r = mid;
else l = mid + 1;
}
flagl = l;
//不存在这样的端点
if (index.get(flagl) > right || index.get(flagl) < left) return 0;
//找第二个
l = 0;
r = n - 1;
int flagr = -1;
while (l < r) {
int mid = l + r >> 1;
if (index.get(mid) >= right) r = mid;
else l = mid + 1;
}
flagr = l;
//⭐找到的右边界是大于或者等于right的第一个数!!因此需要判断
if (index.get(flagr) > right) flagr--;
return flagr - flagl + 1;
}
}