目录
暂无数据
抱歉,暂无数据

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