Nice SubArray

Author: samyam81

June22/NiceSubarray.java

@@ -0,0 +1,16 @@
+public class NiceSubarray{
+ public int numberOfSubarrays(int[] nums, int k) {
+ int n = nums.length;
+ int[] cnt = new int[n + 1];
+ cnt[0] = 1;
+ int ans = 0, t = 0;
+ for (int v : nums) {
+ t += v & 1;
+ if (t - k >= 0) {
+ ans += cnt[t - k];
+ }
+ cnt[t]++;
+ }
+ return ans;
+ }
+}