leetcode刷题日记(697.数组的度)
发表于 2021-03-23 | 最后更新于 2021-12-19 | 开发
给定一个非空且只包含非负数的整数数组 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 范围内的整数。
@芜桐的解法
class Solution:
def findShortestSubArray(self, nums: List[int]) -> int:
mp = {}
for i, item in enumerate(nums):
if item in mp:
mp[item]["c"] += 1
mp[item]["e"] = i
else:
mp[item] = {
"c": 1,
"s": i,
"e": i,
}
# 遍历 mp 找出 count 最大 且 shortest_length 最小的元素,输出 shortest_length
item = list(mp.values())[0]
c = item["c"]
ans = item["e"] - item["s"] + 1
for v in mp.values():
if v["c"] > c or (v["c"] == c and v["e"] - v["s"] + 1 < ans):
c = v["c"]
ans = v["e"] - v["s"] + 1
return ans
许可协议:
CC BY 4.0