fork 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), prefix(n + 1, 0);
  11. for(int i = 0; i < n; ++i) {
  12. cin >> v[i];
  13. prefix[i + 1] = prefix[i] + v[i]; // prefix sum
  14. }
  15.  
  16. int c = 0;
  17. int i = 0;
  18. while (i + k <= n) {
  19. int sum = prefix[i + k] - prefix[i]; // sum of k elements from i
  20. if (sum > 0) c--;
  21. else c++;
  22. i += k + 1; // skip next element
  23. }
  24.  
  25. cout << c << '\n';
  26. }
  27. return 0;
  28. }
  29.  
Success #stdin #stdout 0.01s 5316KB
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