fork download
  1. import java.util.HashMap;
  2. import java.util.Map;
  3. import java.util.Scanner;
  4.  
  5. public class Main {
  6. public static void main(String[] args) {
  7. Scanner scanner = new Scanner(System.in);
  8.  
  9. int n = scanner.nextInt();
  10. long k = scanner.nextLong();
  11. long sum = 0, count = 0;
  12.  
  13. Map<Long, Long> prefixSums = new HashMap<>();
  14. prefixSums.put(0L, 1L);
  15.  
  16. for (int i = 0; i < n; i++) {
  17. sum += scanner.nextLong();
  18. count += prefixSums.getOrDefault(sum - k, 0L);
  19. prefixSums.put(sum, prefixSums.getOrDefault(sum, 0L) + 1);
  20. }
  21.  
  22. System.out.println(count);
  23. }
  24. }
  25.  
Success #stdin #stdout 0.17s 54512KB
stdin
3
5
5 -5 5 
stdout
3