Leetcode 5186：区间内查询数字的频率

题目描述

请你设计一个数据结构，它能求出给定子数组内一个给定值的 频率 。

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

请你实现 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 次。
 

提示：

1 <= arr.length <= 105
1 <= arr[i], value <= 104
0 <= left <= right < arr.length
调用 query 不超过 105 次。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-frequency-queries
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

class RangeFreqQuery {
public:
    unordered_map<int, vector<int>> mp;
    RangeFreqQuery(vector<int>& arr) {
        for(int i = 0; i < arr.size(); i++) {
            int tmp = arr[i];
            if(mp.find(tmp) == mp.end()) mp[tmp] = {};
            mp[tmp].push_back(i);
        }
    }
    
    int query(int left, int right, int value) {
        if(mp.find(value) == mp.end()) return 0;
        const auto &vect = mp[value];
        if(vect[0]>right || vect[vect.size()-1]<left) return 0;
        int l = lower_bound(vect.begin(), vect.end(), left) - vect.begin();
        int r = upper_bound(vect.begin(), vect.end(), right) - vect.begin();
        if(r >= vect.size() || vect[r] > right) r--;
        return (l>r)?0:(r-l+1);
    }
};