Software Engineering/Algorithm
[LeetCode] #1248 Count Number of Nice Subarrays
devhrkim
2024. 8. 5. 08:20
풀이
class Solution {
public int numberOfSubarrays(int[] nums, int k) {
HashMap<Integer, Integer> hashMap = new HashMap<>();
int curr = 0;
int answer = 0;
hashMap.put(0,1);
for(int num: nums) {
curr += num%2;
answer += hashMap.getOrDefault(curr - k, 0);
hashMap.put(curr, hashMap.getOrDefault(curr, 0)+1);
}
return answer;
}
}