K 분할 요소 하위 배열

정수 배열 nums과 두 개의 정수 k 및 p가 주어지면 k로 나눌 수 있는 최대 p개의 요소를 갖는 개별 하위 배열의 수를 반환합니다.

두 배열 nums1 및 nums2는 다음과 같은 경우 구별된다고 합니다.

길이가 다르거나

인덱스i가 하나 이상 존재합니다. 여기서 nums1[i] != nums2[i] 입니다.

하위 배열은 배열에 있는 비어 있지 않은 연속 요소 시퀀스로 정의됩니다.

예 1:

입력: 숫자 = [2,3,3,2,2], k = 2, p = 2
출력: 11
설명:
인덱스 0, 3 및 4의 요소는 p = 2로 나눌 수 있습니다.
2로 나누어지는 최대 k = 2개의 요소를 갖는 11개의 개별 하위 배열은 다음과 같습니다.
[2], [2,3], [2,3,3], [2,3,3,2], [3], [3,3], [3,3,2], [3,3 ,2,2], [3,2], [3,2,2], [2,2].
하위 배열 [2] 및 [3]은 숫자에서 두 번 이상 발생하지만 각각 한 번만 계산되어야 합니다.
하위 배열 [2,3,3,2,2]에는 2로 나누어지는 요소가 3개 있으므로 계산하지 않아야 합니다.

예 2:

입력: 숫자 = [1,2,3,4], k = 4, p = 1
출력: 10
설명:
nums의 모든 요소는 p = 1로 나누어집니다.
또한 숫자의 모든 하위 배열에는 1로 나누어지는 최대 4개의 요소가 있습니다.
모든 하위 배열은 고유하므로 모든 제약 조건을 충족하는 총 하위 배열 수는 10입니다.

제약:

1 <= nums.length <= 200

1 <= nums[i], p <= 200

1 <= k <= nums.length

해결책:

class Solution:
    def countDistinct(self, nums: List[int], k: int, p: int) -> int:
        subs = set()
        left, right, divs = 0, 0, 0
        while right < len(nums):
            if nums[right] % p == 0:
                divs += 1
            while divs > k and left < right:
                if nums[left] % p == 0:
                    divs -= 1
                left += 1
            for l in range(left, right + 1):
                subs.add(tuple(nums[l:right + 1]))
            right += 1
        return len(subs)