fork download
  1. /* package whatever; // don't place package name! */
  2.  
  3. import java.util.*;
  4. import java.lang.*;
  5. import java.io.*;
  6.  
  7. /* Name of the class has to be "Main" only if the class is public. */
  8. class Ideone
  9. {
  10. public static void main (String[] args) throws java.lang.Exception
  11. {
  12. // your code goes here
  13. Scanner sc=new Scanner(System.in);
  14. int n=sc.nextInt();
  15. int k=sc.nextInt();
  16. int a[]=new int[n];
  17. int prefix[]=new int[n];
  18. for(int i=0;i<n;i++)
  19. a[i]=sc.nextInt();
  20.  
  21. prefix[0]=a[0];
  22. for(int i=1;i<n;i++)
  23. prefix[i]=a[i]+prefix[i-1];
  24.  
  25. Map<Integer,Integer> m=new HashMap<>();
  26. m.put(0,1);
  27. int count=0;
  28. for(int ele:prefix)
  29. {
  30. int cpl=ele-k;
  31. if(m.containsKey(cpl))
  32. count+=m.get(cpl);
  33. m.put(ele,m.getOrDefault(ele,0)+1);
  34. }
  35.  
  36. System.out.println(count);
  37.  
  38. }
  39. }
Success #stdin #stdout 0.12s 54636KB
stdin
5
5
1 2 3 4 5
stdout
2