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


这一题的大意是让我们寻找在一个数组中的某一子数组中某一个元素出现的次数。

找一个元素在某一数组中出现的次数,我们通常会想到的是遍历一遍数组用哈希表来统计该元素出现的次数,但问题是这道题中会有最多10^5次询问,而数组的长度最多为 10 ^5,暴力遍历肯定是超时的,这时候我们可以采用二分的形式,通过找该元素最早出现的位置和该元素最晚出现的位置,两者相减就是出现的次数,时间复杂度O(nlogn)可以接受

但是要注意的是虽然使用二分可以下降到O(logn)但是二分需要在单调区间中才能使用,而题目中给出的序列是无序的,我们如果在每一次进行二分前进行一个排序的话,那么时间复杂度会超时,O(n)*(nlogn)(排序的时间复杂度 * n次询问)。

我之前写的就是排序+二分的做法超时了:

cpp 复制代码
class RangeFreqQuery {
public:
    vector<int> a;
    RangeFreqQuery(vector<int>& arr) {
        for(int i=0;i<arr.size();i++)
        {
            a.push_back(arr[i]);
        }
    }
    int query(int left, int right, int value) {
            //排序
            vector<int> temp=a;
            sort(temp.begin()+left,temp.begin()+right+1);
            int l=left;
            int r=right;
            //找第一个大于等于value的位置
            //找第一个小于等于value的位置
            //如果数组中有value 那么
            // 那么这两个位置都一定在value上
            int lx=INT_MAX;
            while(l<=r)
            {
              int mid=(l+r)/2;
              if(temp[mid]==value)
              {
                 r=mid-1;
                 lx=min(lx,mid);
              }
              else if(temp[mid]>value)
              {
                r=mid-1;
              }
              else{
                l=mid+1;
              }
            } 
            //找到最左端的lx value
             int rx=-1;
             l=left;
             r=right;
             while(l<=r)
             {
                int mid=(l+r)/2;
                if(temp[mid]==value)
                {
                    l=mid+1;
                    rx=max(rx,mid);
                }
                else if(temp[mid]>value){
                    r=mid-1;
                }
                else{
                    l=mid+1;
                }
             }
             //这样就找到了lx和lr的位置了
             if(lx==INT_MAX||rx==-1)
             {
                return 0;
             }
             else 
             {
                return rx-lx+1;
             }
    }
};

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

实际应该用哈希+二分,我们在询问之前,就用哈希表统计出每一个值在数组中可能出现的位置,然后我们用一个二分来看可能存在的位置是否在询问的区间内,分别找到出现在区间内的最小位置和最大位置,二者之差即为出现的频率次数。

完整的代码如下:

cpp 复制代码
class RangeFreqQuery {
public:
    vector<int> a;
    unordered_map<int,vector<int> > mp;
    RangeFreqQuery(vector<int>& arr) {
        for(int i=0;i<arr.size();i++)
        {
            a.push_back(arr[i]);
            mp[arr[i]].push_back(i);
        }
    }
    int query(int left, int right, int value) {
            //排序
            vector<int>& temp=mp[value];
            if(mp[value].size()==0)
            {
            return 0;     
            }
            int l=0;
            int r=temp.size()-1;
            //找第一个大于等于value的位置
            //找第一个小于等于value的位置
            //如果数组中有value 那么
            // 那么这两个位置都一定在value上
            int lx=INT_MAX;
            while(l<=r)
            {
              int mid=(l+r)/2;
              if(temp[mid]>=left&&temp[mid]<=right)
              {
                 r=mid-1;
                 lx=min(lx,mid);
              }
              else if(temp[mid]>right)
              {
                r=mid-1;
              }
              else{
                l=mid+1;
              }
            } 
            //找到最左端的lx value
             int rx=-1;
             l=0;
             r=temp.size()-1;
             while(l<=r)
             {
                int mid=(l+r)/2;
                if(temp[mid]>=left&&temp[mid]<=right)
                {
                    l=mid+1;
                    rx=max(rx,mid);
                }
                else if(temp[mid]>right){
                    r=mid-1;
                }
                else{
                    l=mid+1;
                }
             }
             //这样就找到了lx和lr的位置了
             if(lx==INT_MAX||rx==-1)
             {
                return 0;
             }
             else 
             {
                return rx-lx+1;
             }
    }
};

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

需要注意的是,在获取哈希表中的某一个值对应的出现位置的数组时,我们要用引用来获取,否则会超时

cpp 复制代码
vector<int>& temp=mp[value];

最终的时间复杂度是O(nlogn)

相关推荐
民乐团扒谱机25 分钟前
【微实验】NSGA-II(非支配排序遗传算法)——解决多目标优化问题的“自然选择模拟器”
linux·人工智能·算法
青少儿编程课堂34 分钟前
蓝桥杯·全国青少年信息素养大赛·电子学会Python/C++考级:中缀表达式求值解析
c++·python·算法·bfs·信息学竞赛
文丑颜不良啊1 小时前
LeetCode2261~2280题目解析
java·leetcode
武子康3 小时前
Qwen-Drive 到底是什么:4B 级 VLM 如何成为驾驶策略底座
人工智能·算法·机器学习
zh_xuan4 小时前
LeeCode 496. 下一个更大元素I
数据结构·算法·栈·单调栈
机 _ 长4 小时前
05 智算运维|8 张卡怎么才算「捆成一张」:NCCL 实测与拓扑
运维·人工智能·算法·智算运维
早起CaiCai5 小时前
【2023-01 OM】利用深度学习算法融合异质观测实现海洋场降尺度
人工智能·深度学习·算法
梦里寻码6 小时前
我用 NestJS + Vue3 写了一套完整的研发管理系统(58 表 / 229 API / 130 项断言,全开源
前端·后端·算法
猿人谷6 小时前
Surrogate Model:让进化算法学会“预测”,为什么它正在成为智能优化的重要技术?
后端·算法·架构
文丑颜不良啊7 小时前
LeetCode2241~2260题目解析
java·leetcode