LeetCode 1296. 劃分數組為連續數字的集合

題目

給你一個整數數組 nums 和一個正整數 k，請你判斷是否可以把這個數組劃分成一些由 k 個連續數字組成的集合。

如果可以，請傳回 True；否則，傳回 False。

示例 1：

輸入：nums = [1,2,3,3,4,4,5,6], k = 4
輸出：true
解釋：數組可以分成 [1,2,3,4] 和 [3,4,5,6]。
示例 2：

輸入：nums = [3,2,1,2,3,4,3,4,5,9,10,11], k = 3
輸出：true
解釋：數組可以分成 [1,2,3] , [2,3,4] , [3,4,5] 和 [9,10,11]。
示例 3：

輸入：nums = [3,3,2,2,1,1], k = 3
輸出：true
示例 4：

輸入：nums = [1,2,3,4], k = 3
輸出：false
解釋：數組不能分成幾個大小為 3 的子數組。           

提示：

1 <= nums.length <= 10^5

1 <= nums[i] <= 10^9

1 <= k <= nums.length

解題思路

class Solution:
    def isPossibleDivide(self, nums: [int], k: int) -> bool:
        #用remove實作
        # if len(nums)%k != 0: return False
        # nums.sort()
        # while len(nums) != 0:
        #     tempNum = nums[0]
        #     for i in range(k):
        #         try:
        #             nums.remove(tempNum+i)
        #         except:
        #             return False
        # return True
        #用Counter内置函數實作
        if len(nums)%k != 0: return False
        nums.sort()
        import collections
        counter = collections.Counter(nums)
        counterNum = 0
        for num in nums:
            if counter[num] == 0:continue
            for i in range(k):
                counter[num+i] -= 1
                if counter[num+i] < 0:return False
                counterNum += 1
        if counterNum != len(nums):return False
        return True