leetcode每日刷题:数组的度
题目描述:给定一个非空且只包含非负数的整数数组 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
解题思路:
实际上这个题目就是要找到给定列表里面出现频率最高的数字,然后返回最后一次出现的位置减去最初出现的位置,
不过要注意的是,如果有多个出现频率相同的数字,要返回最小的长度
所以,可以使用一个字典来对每个数字出现的频率,第一次出现的位置,最后一次出现的位置进行保存
最后,查一下这个字典里面频率最高的,而且最后一次出现的位置减去第一次出现的位置的最小值就是最后的结果。
class Solution: def findShortestSubArray(self, nums: List[int]) -> int: #用来保存出现的次数,第一次出现的位置,最后一次出现的位置 mp = dict() for i,num in enumerate(nums): if num in mp: mp[num][0]+=1 mp[num][2] =i else: mp[num]=[1,i,i] max_num=0 min_len=0 #对字典进行检索 for count,left,right in mp.values(): #如果当前的数字的频率比max_num高 if max_num<count: max_num = count min_len = right - left + 1 #如果当前数字的频率和max_num一样,这时候就比较一下最短长度看谁短 elif max_num==count: if min_len>(right-left+1): min_len = right-left+1 return min_len
下一篇:
冒泡排序算法的范型实现