1
0
Fork 0
mirror of https://gitlab.com/mfocko/LeetCode.git synced 2024-09-19 17:56:55 +02:00
LeetCode/go/count-number-of-nice-subarrays.go

24 lines
349 B
Go
Raw Normal View History

package main
func numberOfSubarrays(nums []int, k int) int {
atMost := func(k int) int {
subarrays := 0
window, start := 0, 0
for end := range nums {
window += nums[end] % 2
for window > k {
window -= nums[start] % 2
start++
}
subarrays += end - start + 1
}
return subarrays
}
return atMost(k) - atMost(k-1)
}