1. 程式人生 > >leetcode 697. 陣列的度(python)

leetcode 697. 陣列的度(python)

給定一個非空且只包含非負數的整數陣列 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)