【题目描述】
给定一个非空且只包含非负数的整数数组 nums, 数组的度的定义是指数组里任一元素出现频数的最大值。
你的任务是找到与 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.
输入: [1,2,2,3,1,4,2]
输出: 6
【解题思路】
1.最开始的思路就是先遍历找到最大频数,然后再遍历找到所有满足最大频数的数字,再算出每个数字出现的最左最右下标,然后相减比较最小值,这种虽然通过了,但是耗时挺多的,代码看着也很复杂。
2.借鉴了别人的解法,就是存储每个数字出现的index就不用向上面解法频繁的遍历数组
【代码】
//思路一
class Solution {
public:
int findShortestSubArray(vector<int>& nums) {
unordered_map<int, int>table;
for(int i = 0; i < nums.size(); i++)
table[nums[i]]++;
unordered_map<int, int>::iterator it;
int cnt = INT_MIN;
for(it = table.begin(); it !=table.end(); it++)
{
cnt = max(cnt, it->second);
}
vector<int>res;
for(it = table.begin(); it !=table.end(); it++)
{
if(it->second == cnt)
res.push_back(it->first);
}
int len = INT_MAX;
for(int i = 0; i < res.size(); i++)
{
int first = 0;
while(first < nums.size() && nums[first] != res[i])
first++;
int second = first;
for(int j = first+1; j < nums.size(); j++)
{
if(nums[j] == res[i])
second = j;
}
len = min(len, second-first+1);
}
return len;
}
};
//思路二
class Solution {
public:
int findShortestSubArray(vector<int>& nums) {
unordered_map<int, vector<int>>table;
for(int i = 0; i < nums.size(); i++)
table[nums[i]].push_back(i);
int du = 1;
int len = INT_MAX;
unordered_map<int, vector<int>>::iterator it;
for(it = table.begin(); it !=table.end(); it++)
{
if(it->second.size() > du)
{
du = it->second.size();
len = it->second.back()-it->second[0] +1;
}
else if(it->second.size() == du)
len = min(len, it->second.back()-it->second[0] +1);
}
return len;
}
};