目錄
一、題目內容
二、解題思路
三、代碼
一、題目內容
給定一個非空且只包含非負數的整數陣列 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:
nums_dict = {}
for i in range(len(nums)):
if nums[i] not in nums_dict:
nums_dict[nums[i]] = [i]
else:
nums_dict[nums[i]].append(i)
max_length = 0
for num in nums_dict:
max_length = max(max_length, len(nums_dict[num]))
min_length = len(nums)
for num in nums_dict:
if len(nums_dict[num]) == max_length:
min_length = min(min_length, nums_dict[num][-1] - nums_dict[num][0] + 1)
return min_length
if __name__ == '__main__':
s = Solution()
nums = [1,2,2,1,2,1,1,1,1,2,2,2]
ans = s.findShortestSubArray(nums)
print(ans)
CSDN認證博客專家
深度學習
神經網路
Pytorch
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/261734.html
標籤:python
上一篇:python爬蟲入門(2)
