开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第20天,点击查看活动详情.
一、题目描述:
给定一个非空且只包含非负数的整数数组 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 范围内的整数。
二、思路分析:
- 用map统计各个元素的频数。取频数最高的元素。
- 在数组中找到此元素最左值和最右值,即下标。
- 然后下标相减,即最小长度。选出最小的长度。
三、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;
}
};