给定一个非空且只包含非负数的整数数组 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
提示:
- nums.length 在1到 50,000 区间范围内。
- nums[i] 是一个在 0 到 49,999 范围内的整数。
解题思路:
又是一遭写代码遇bug…奔向了题解
- 创建字典,统计每个元素出现的频率。统计时,value值可用数组来保存该元素在原数组中的下标。
- 根据每个元素出现的频率进行排序,从小到大
len(value)
- 遍历排序后的键值对,以元素出现频率作为判定标准:如果是最大出现频率,则更新子数组长度;否则,可直接退出循环。
代码:
class Solution:
def findShortestSubArray(self, nums: List[int]) -> int:
dic = {}
for i in range(len(nums)):
if nums[i] not in dic:
dic[nums[i]] = [i]
else:
dic[nums[i]].append(i)
key_value = sorted(dic.items(), key = lambda x: -len(x[1]))
count = len(key_value[0][1])
res = len(nums)
for j in key_value:
if len(j[1]) == count:
res = min(res, j[1][count-1] - j[1][0] + 1)
else:
break
return res
提交记录: