LeetCode 697. 数组的度
天津科技大学第六届科技文化节算法设计大赛第13题
难度:中等
题目:
给定一个非空且只包含非负数的整数数组 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 范围内的整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/degree-of-an-array
解题思路:
- 将字符串转成集合以获得含有唯一字母的集合
- 求出每个字母的度
- 定义一函数,可分别从头和尾查找到元素下标(元素第一次出现的下标为low,最后一次出现的下标为high)
- 则与 nums 拥有相同大小的度的最短连续子数组的长度为distance=high-low+1
- 寻找最短的distance
源代码
def findShortestSubArray(self, nums: List[int]) -> int:
setnum=set(nums)
duli=[]
for i in setnum:
duli.append(nums.count(i)) #父数组的度集合
du=max(duli) #父数组的度
dumax=[]#等于父数组度的元素集合
for i in setnum:
if nums.count(i)==du:#若i的度为父数组的度
dumax.append(i)
min=100000
def findelement(element,alist,order):
if order==1:
return alist.index(element)
elif order==0:
for i in range(len(alist)-1,-1,-1):
if alist[i]==element:
return i
for i in dumax:
low=findelement(i,nums,1)
high=findelement(i,nums,0)
distance=high-low+1
if distance<min:
min=distance
return min