fork(1) download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. int t;
  6. cin >> t;
  7. while(t--) {
  8. int n, k;
  9. cin >> n >> k;
  10. vector<int> v(n);
  11. for(int i = 0; i < n; ++i)
  12. cin >> v[i];
  13.  
  14. int c = 0;
  15. int i = 0;
  16.  
  17. while (i + k <= n) {
  18. int sum = 0;
  19. for (int j = 0; j < k; ++j)
  20. sum += v[i + j];
  21.  
  22. if (sum > 0)
  23. c--;
  24. else
  25. c++;
  26.  
  27. i += k + 1; // move window by k elements, then skip 1
  28. }
  29.  
  30. cout << c << '\n';
  31. }
  32. return 0;
  33. }
  34.  
Success #stdin #stdout 0s 5308KB
stdin
5
5 1
0 1 0 0 0
7 3
0 0 0 0 0 0 0
3 1
1 1 1
4 2
0 1 0 1
6 2
0 0 1 0 0 0
stdout
3
2
-2
-1
2