fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. void solve() {
  5. int n, k;
  6. cin >> n >> k;
  7. string s;
  8. cin >> s;
  9.  
  10. int z = count(s.begin(), s.end(), '0');
  11. int P = n / 2;
  12.  
  13. // difference from perfect half
  14. int diff = abs(z - P);
  15.  
  16. // parity check: (z + k - P) even ⟺ (k - diff) even
  17. bool ok = (diff <= k && k <= P && ((k - diff) % 2 == 0));
  18.  
  19. cout << (ok ? "YES\n" : "NO\n");
  20. }
  21.  
  22. int main(){
  23. ios::sync_with_stdio(false);
  24. cin.tie(nullptr);
  25.  
  26. int t;
  27. cin >> t;
  28. while (t--) solve();
  29. return 0;
  30. }
  31.  
Success #stdin #stdout 0.01s 5328KB
stdin
6
6 2
000000
2 1
01
4 1
1011
10 2
1101011001
10 1
1101011001
2 1
11
stdout
NO
NO
YES
NO
YES
YES