Leetcode.2080 区间内查询数字的频率

最新推荐文章于 2024-07-16 18:35:54 发布

感觉画质不如…原神

最新推荐文章于 2024-07-16 18:35:54 发布

阅读量570

点赞数 3

分类专栏： Leetcode 文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/m0_74396439/article/details/129102141

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

题目链接

Leetcode.2080 区间内查询数字的频率 Rating ： 1702

题目描述

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

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

请你实现 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 <= 10^5$
$1 <= arr[i], value <= 10^4$
$0 <= l e f t <= r i g h t < a rr . l e n g t h$
调用 query不超过 $10^5$ 次。

分析：

如果暴力求解，时间复杂度是 $O (n * q)$ ，最高是 $10^{10}$ （一般的oj，1s的计算量一般为 $10^8$ ）,这是肯定会超时的。

我们用一个哈希表，记录每一个元素x对应的下标。

对于 query(int left, int right, int value)，我们只需要通过二分的方式计算 value的下标有多少个是在区间 [left,right]中的。

时间复杂度： $O (q * l o g n)$

C++代码：

class RangeFreqQuery {
private: 
    // 数值为键，出现下标数组为值的哈希表
    unordered_map<int, vector<int>> occurence;
    
public:
    RangeFreqQuery(vector<int>& arr) {
        // 顺序遍历数组初始化哈希表
        int n = arr.size();
        for (int i = 0; i < n; ++i){
            occurence[arr[i]].push_back(i);
        }
    }
    
    int query(int left, int right, int value) {
        auto &pos = occurence[value];
        //计算大于等于 left的位置
        auto l = lower_bound(pos.begin(), pos.end(), left);
        //计算 大于right的位置
        auto r = upper_bound(pos.begin(), pos.end(), right);
        
        return r - l;
    }
};

Java代码：

class RangeFreqQuery {
    Map<Integer,List<Integer>> map = new HashMap<>();

    public RangeFreqQuery(int[] arr) {
        int n = arr.length;
        for(int i = 0;i < n;i++){
            int x = arr[i];
            if(map.containsKey(x)) map.get(x).add(i);
            else{
                map.put(x,new ArrayList<>());
                map.get(x).add(i);
            }
        }
    }
    
    public int query(int left, int right, int value) {
        if(!map.containsKey(value)) return 0;

        List<Integer> list = map.get(value);

        int n = list.size();
        //求 第一个大于等于 left 的下标 i
        int l = 0,r = n - 1;
        while(l < r){
            int mid = (l + r) >> 1;
            if(list.get(mid) >= left) r = mid;
            else l = mid + 1;
        }
        //不满足要求直接返回0
        if(list.get(l) < left) return 0;
        int i = l;

        //求第一个 小于等于 right 的下标 j
        l = 0;
        r = n - 1;
        
        while(l<r){
            int mid = (l + r + 1)>>1;
            if(list.get(mid) <= right) l = mid;
            else r = mid - 1;
        }
        if(list.get(r) > right) return 0;
        int j = r;

        return j - i + 1;
    }
}

/**
 * Your RangeFreqQuery object will be instantiated and called as such:
 * RangeFreqQuery obj = new RangeFreqQuery(arr);
 * int param_1 = obj.query(left,right,value);
 */