fork download
  1. import java.util.*;
  2.  
  3. class Ideone {
  4. public static void main(String[] args) {
  5. Scanner sc = new Scanner(System.in);
  6. int n = sc.nextInt();
  7. int k = sc.nextInt();
  8. int arr[] = new int[n];
  9. for (int i = 0; i < n; i++) arr[i] = sc.nextInt();
  10. HashMap<Integer,Integer> hm = new HashMap<>();
  11. hm.put(0,1);
  12. int sum = 0, count = 0;
  13. for (int i = 0; i < n; i++) {
  14. sum += arr[i];
  15. if (hm.containsKey(sum - k)) count += hm.get(sum - k);
  16. hm.put(sum, hm.getOrDefault(sum,0)+1);
  17. }
  18. System.out.println(count);
  19. }
  20. }
Success #stdin #stdout 0.18s 56704KB
stdin
3
5
2
3
5
stdout
2