题目描述
请你设计一个数据结构,它能求出给定子数组内一个给定值的 频率 。
子数组中一个值的 频率 指的是这个子数组中这个值的出现次数。
请你实现 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]);
(1, 2, 4); // 返回 1 。4 在子数组 [33, 4] 中出现 1 次。
(0, 11, 33); // 返回 2 。33 在整个子数组中出现 2 次。
提示:
1 <= <= 105
1 <= arr[i], value <= 104
0 <= left <= right <
调用 query 不超过 105 次。
解题思路
class RangeFreqQuery {
public:
unordered_map<int, vector<int>> mp;
RangeFreqQuery(vector<int>& arr) {
for(int i = 0; i < (); i++) {
int tmp = arr[i];
if((tmp) == ()) mp[tmp] = {};
mp[tmp].push_back(i);
}
}
int query(int left, int right, int value) {
if((value) == ()) return 0;
const auto &vect = mp[value];
if(vect[0]>right || vect[()-1]<left) return 0;
int l = lower_bound((), (), left) - ();
int r = upper_bound((), (), right) - ();
if(r >= () || vect[r] > right) r--;
return (l>r)?0:(r-l+1);
}
};