LeetCode 697. 数组的度]

84 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第20天,点击查看活动详情.

一、题目描述:

697. 数组的度

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

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

示例 1:

输入:nums = [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:

输入:nums = [1,2,2,3,1,4,2]
输出:6
解释:
数组的度是 3 ,因为元素 2 重复出现 3 次。
所以 [2,2,3,1,4,2] 是最短子数组,因此返回 6 。

  提示:

  • nums.length 在 1 到 50,000 范围内。
  • nums[i] 是一个在 0 到 49,999 范围内的整数。

二、思路分析:

  1. 用map统计各个元素的频数。取频数最高的元素。
  2. 在数组中找到此元素最左值和最右值,即下标。
  3. 然后下标相减,即最小长度。选出最小的长度。

三、AC 代码:

class Solution {
public:
    int findShortestSubArray(vector<int>& nums) {
        map<int, int> m;
        int maxNum = 0;
        //找到频数最多的个数
        for (int i = 0; i < nums.size(); i++) {
            m[nums[i]]++;
            maxNum = max(maxNum, m[nums[i]]);
        }
        vector<int> heighNum;
        map<int, int>::iterator itr;
        //找到频数最多的元素
        for (itr = m.begin(); itr != m.end(); itr++) {
            if (itr->second == maxNum) {
                heighNum.push_back(itr->first);
            }
        }
        vector<vector<int>> saveIndex(heighNum.size(),vector<int>(2));
        //遍历频度最高的噪声,找到最小子数组左右下标
        for (int i = 0; i < heighNum.size(); i++) {
            int indexNum = 0;
            for (int j = 0; j < nums.size(); j++) {
                if (nums[j] == heighNum[i] && indexNum == 0) {
                    saveIndex[i][0] = j;
                    indexNum++;
                }
                if (nums[j] == heighNum[i] && indexNum != 0) {
                    saveIndex[i][1] = j>saveIndex[i][1]?j: saveIndex[i][1];
                }
            }
        }
        //比较最小数组左右下标,选取最小的下标差+1
        int inial = saveIndex[0][1] - saveIndex[0][0];
        for (int j = 1; j < saveIndex.size(); j++) {
            inial = (saveIndex[j][1] - saveIndex[j][0]) < inial ? (saveIndex[j][1] - saveIndex[j][0]) : inial;
        }
        return inial+1;
    }
};

范文参考:

custer学习笔记 - 数组的度 - 力扣(LeetCode) (leetcode-cn.com)