题目:
697、数组的度
给定一个非空且只包含非负数的整数数组 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 范围内的整数。
题解思路:
1、哈希表每个key设置为nums[i]的值,values设置为该值的索引值,方便计算度的长度。
2、找到数组的度,然后找到和度大小一致的出现的元素,根据元素出现的首末位置计算最短子数组长度。
注意:和度大小一致的出现的元素可能不止一个,需要加上循环更新。另外,最短子数组长度一定为该元素出现的首末位置索引之差
注意字典类型定义时,需要使用collection.defaultdict(list)函数,因为普通dict()不允许d[].append未预定义key的添加操作。
题解python代码:
class Solution:
def findShortestSubArray(self, nums: List[int]) -> int:
d = collections.defaultdict(list)
for i in range(len(nums)):
d[nums[i]].append(i) # 哈希表每个key设置为nums[i]的值,values设置为该值的索引值,方便计算度的长度
# 按出现次数排序
t = sorted(d, key=lambda x: len(d[x]), reverse=True)
# 确定度
degree = len(d[t[0]])
# 找到满足度为degree的最短子数组
ans = 50000
for i in t: # 设置这个循环是为了排序有多个出现都为degree的值,如示例1中的1和2
if len(d[i])==degree:
ans = min(ans, d[i][-1]-d[i][0]+1)
else:
break
# print(t, d)
return ans
作者:a-qing-ge
链接:https://leetcode-cn.com/problems/degree-of-an-array/solution/ha-xi-biao-su-du-hen-kuai-by-a-qing-ge-vme9/
来源:力扣(LeetCode)https://leetcode-cn.com/problems/degree-of-an-array/