力扣打卡2021.2.20 数组的度

最新推荐文章于 2024-10-08 20:32:22 发布

Freddie_sw

最新推荐文章于 2024-10-08 20:32:22 发布

阅读量110

点赞数

文章标签：算法 leetcode

本文链接：https://blog.csdn.net/weixin_45780132/article/details/113899658

版权

该博客介绍了一个算法问题，即在非空且非负的整数数组中找到度相同的最短连续子数组。度定义为数组中元素的最大出现频率。代码实现中，使用了哈希映射记录每个元素的出现次数和位置，找到最大频率及其对应的元素，最后找出最短连续子数组的长度。示例展示了输入和输出，并解释了代码的工作原理。

摘要由CSDN通过智能技术生成

题目：
给定一个非空且只包含非负数的整数数组 nums，数组的度的定义是指数组里任一元素出现频数的最大值。

你的任务是在 nums 中找到与 nums 拥有相同大小的度的最短连续子数组，返回其长度。

示例 1：

输入：[1, 2, 2, 3, 1]
输出：2
解释：
输入数组的度是2，因为元素1和2的出现频数最大，均为2.
连续子数组里面拥有相同度的有如下所示:
[1, 2, 2, 3, 1], [1, 2, 2, 3], [2, 2, 3, 1], [1, 2, 2], [2, 2, 3], [2, 2]
最短连续子数组[2, 2]的长度为2，所以返回2.
示例 2：

输入：[1,2,2,3,1,4,2]
输出：6

代码：

class Solution {
public:
    int findShortestSubArray(vector<int>& nums) {
        if(nums.size()<=1) return nums.size();

        int mp[50000][3];
        vector<int> max_fn;
        int max_f=0;

        for(int i=0;i<50000;i++){
            mp[i][0]=0;
            mp[i][1]=-1;
            mp[i][2]=-1;
        }


        for(int i=0;i<nums.size();i++){
            if(mp[nums[i]][0]==0){
                mp[nums[i]][1]=i;
            }
            mp[nums[i]][2]=i;
            mp[nums[i]][0]++;

            if(mp[nums[i]][0]>=max_f){
                max_f=mp[nums[i]][0];
                if(max_fn.size()>0 && mp[max_fn[max_fn.size()-1]][0]<max_f || max_fn.size()>0 && max_fn[max_fn.size()-1]==nums[i]){
                    while(max_fn.size()>0){
                        max_fn.pop_back();
                    }
                }
                max_fn.push_back(nums[i]);
            }
        }

        int res=50000;
        int tmp;
        for(int i=0;i<max_fn.size();i++){
            tmp=mp[max_fn[i]][2]-mp[max_fn[i]][1];
            if(tmp<res){
                res=tmp;
            }
        }
        return res+1;
    }
};