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[] nums=new int[n+1];
  16. int totalOneCount=0;
  17. for(int i=1;i<=n;i++){
  18. nums[i]=sc.nextInt();
  19. if(nums[i]==1) totalOneCount++;
  20. }
  21. int k=sc.nextInt();
  22. int zeros = 0;
  23. for (int i = 0; i < k; i++) {
  24. if (nums[i] == 0) zeros++;
  25. }
  26. int maxZeros = zeros;
  27.  
  28. for (int i = k; i < n; i++) {
  29. if (nums[i - k] == 0) zeros--;
  30. if (nums[i] == 0) zeros++;
  31. maxZeros = Math.max(maxZeros, zeros);
  32. }
  33. int y = k - maxZeros;
  34. int sum=y*(y+1)/2;
  35. int ans=sum+totalOneCount-y;
  36. System.out.println(ans);
  37.  
  38. }
  39. }
Success #stdin #stdout 0.14s 54668KB
stdin
5 
1 1 1 0 1
4 
stdout
7