You are given an integer array nums
and an integer k
.
A subarray is called prime-gap balanced if:
k
.Return the count of prime-gap balanced subarrays in nums
.
Note:
Example 1:
Input: nums = [1,2,3], k = 1
Output: 2
Explanation:
Prime-gap balanced subarrays are:
[2,3]
: contains two primes (2 and 3), max - min = 3 - 2 = 1 <= k
.[1,2,3]
: contains two primes (2 and 3), max - min = 3 - 2 = 1 <= k
.Thus, the answer is 2.
Example 2:
Input: nums = [2,3,5,7], k = 3
Output: 4
Explanation:
Prime-gap balanced subarrays are:
[2,3]
: contains two primes (2 and 3), max - min = 3 - 2 = 1 <= k
.[2,3,5]
: contains three primes (2, 3, and 5), max - min = 5 - 2 = 3 <= k
.[3,5]
: contains two primes (3 and 5), max - min = 5 - 3 = 2 <= k
.[5,7]
: contains two primes (5 and 7), max - min = 7 - 5 = 2 <= k
.Thus, the answer is 4.
Constraints:
1 <= nums.length <= 5 * 104
1 <= nums[i] <= 5 * 104
0 <= k <= 5 * 104