fork download
  1. #include <iostream>
  2. #include<vector>
  3. using namespace std;
  4.  
  5. void solve() {
  6. int n, x; cin >> n >> x;
  7. vector<int> a(n);
  8. for (int &i : a) cin >> i;
  9.  
  10. for (int i = 0; i < n; ++i) {
  11. if (a[i] == 1) {
  12. int j = i, time = x;
  13. while (j < n && (a[j] == 0 || time > 0)) {
  14. if (a[j] == 1) --time;
  15. ++j;
  16. }
  17. if (j == n) {
  18. cout << "YES\n";
  19. return;
  20. }
  21. }
  22. }
  23. cout << "NO\n";
  24. }
  25.  
  26. int main() {
  27. ios::sync_with_stdio(false);
  28. cin.tie(0);
  29. int t; cin >> t;
  30. while (t--) solve();
  31. }
Success #stdin #stdout 0.01s 5288KB
stdin
7
4 2
0 1 1 0
6 3
1 0 1 1 0 0
8 8
1 1 1 0 0 1 1 1
1 2
1
5 1
1 0 1 0 1
7 4
0 0 0 1 1 0 1
10 3
0 1 0 0 1 0 0 1 0 0
stdout
YES
YES
YES
YES
YES
YES
YES