leetcode 697. 陣列的度(python)
阿新 • • 發佈:2018-11-07
給定一個非空且只包含非負數的整數陣列 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(object): def findShortestSubArray(self, nums): """ :type nums: List[int] :rtype: int """ counts = collections.Counter(nums) left, right = {}, {} #定義兩個字典,存左右邊界 for i, num in enumerate(nums): #如果字典中包含有給定鍵,則返回該鍵對應的值,否則返回為該鍵設定的值。這樣一來取得都是左邊的 left.setdefault(num, i) #for 迴圈過程中,每一次都會更新值,最後一次執行時,right存下的都是最右邊的邊界 right[num] = i degree = max(counts.values()) res = min(right[num]-left[num]+1 for num in counts.keys() if counts[num] == degree) return res
知識點一:enumerate()
enumerate() 函式用於將一個可遍歷的資料物件(如列表、元組或字串)組合為一個索引序列,同時列出資料和資料下標,一般用在 for 迴圈當中。Python 2.3. 以上版本可用,2.6 新增 start 引數。
>>>seq = ['one', 'two', 'three']
>>> for i, element in enumerate(seq):
... print i, element
...
0 one
1 two
2 three
知識點二:setdefault()
Python 字典 setdefault() 函式和 get() 方法類似, 如果鍵不存在於字典中,將會新增鍵並將值設為預設值。如果字典中包含有給定鍵,則返回該鍵對應的值,否則返回為該鍵設定的值。
dict.setdefault(key, default=None)