数组的度
给定一个非空且只包含非负数的整数数组 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 范围内的整数。
解析:
创建一个字典a,用来记录nums中每个元素出现的位置。度为出现位置最多的元素len,创建for循环,求出度并写入m。返回拥有相同度的元素中,求min(首末位置最+1)。
class Solution:
def findShortestSubArray(self, nums: List[int]) -> int:
#用dict记录nums中每个元素出现的位置,
a = {}
for i in range(len(nums)):
if nums[i] not in a:
a[nums[i]] = [i]
else:
a[nums[i]].append(i)
m = 0
#度为出现位置最多的len
for i in a:
m = max(m, len(a[i]))
r = len(nums)
#返回拥有相同度的元素中,min(首末位置最+1)
for i in a:
if len(a[i]) == m:
r = min(r, a[i][-1] - a[i][0] + 1)
return r