fork download
copy 
  1. #include <iostream>
  2. using namespace std;
  3.  
  4. int main(){
  5.     int t; cin >> t;
  6.  
  7.     while(t --){
  8.         int n, k; cin >> n >> k;
  9.         bool used = false;
  10.         bool ok = true;
  11.         int time = 0;
  12.  
  13.         for(int i = 1; i <= n; i ++){
  14.             int x; cin >> x;
  15.             if(used == false && x == 1){
  16.                 used = true;
  17.             }
  18.             if(used == true && time != k){
  19.                 time ++;
  20.             }
  21.             else if(time == k && x == 1){
  22.                 ok = false;
  23.             }
  24.         }
  25.         if(ok == true){
  26.             cout << "YES\n";
  27.         }
  28.         else{
  29.             cout << "NO\n";
  30.         }
  31.     }
  32. }
Success #stdin #stdout 0s 5324KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
YES
YES
https://ideone.com/sU3Y1T
Problem G
language:
C++ (gcc 8.3)
created:
15 hours ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!