aboutsummaryrefslogtreecommitdiff
path: root/count_subarr_max_elem_k.py
diff options
context:
space:
mode:
Diffstat (limited to 'count_subarr_max_elem_k.py')
-rw-r--r--count_subarr_max_elem_k.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/count_subarr_max_elem_k.py b/count_subarr_max_elem_k.py
index 8f734f2..7852d49 100644
--- a/count_subarr_max_elem_k.py
+++ b/count_subarr_max_elem_k.py
@@ -15,18 +15,18 @@ class Solution(object):
:type k: int
:rtype: int
"""
+ ans, curr, l = 0, 0, 0
m = max(nums)
- ans, curr, i = 0, 0, 0
- for j in range(len(nums)):
- curr += nums[j] == m
+ for r in range(len(nums)):
+ curr += nums[r] == m
while curr >= k:
- curr -= nums[i] == m
- i += 1
- ans += i
+ curr -= nums[l] == m
+ l += 1
+ ans += l
return ans
if __name__ == "__main__":
obj = Solution()
- print(obj.countSubarrays(nums=[1, 3, 2, 3, 3], k=2)) # expect: 6
- print(obj.countSubarrays(nums=[1, 4, 2, 1], k=3)) # expect: 0
+ print(obj.countSubarrays(nums=[1, 3, 2, 3, 3], k=2))
+ print(obj.countSubarrays(nums=[1, 4, 2, 1], k=3))