Submission #1114882

#	Time	Username	Problem	Language	Result	Execution time	Memory
1114882		PagodePaiva	COVID tests (CEOI24_covid)	C++17	10 / 100	6098 ms	732712 KiB

Main

#include<bits/stdc++.h>

using namespace std;

vector <int> idx;

  int n, t;


bool query(string s){
  cout << "Q " << s << '\n';
  cout << flush;
  char c;
  cin >> c;
  if(c == 'P')
    return true;
  return false;
}

string solve(string s){
  string s1, s2;
  int con = 0;
  for(int i = 0;i < s.size();i++){
    s1.push_back('0');
    s2.push_back('0');
    if(s[i] == '1') con++;
  }
  if(con == 1) return s;
  int d = 0, p = con/2;
  for(int i = 0;i < s.size();i++){
    if(s[idx[i]] == '1'){
      if(d < p){
        s1[idx[i]] = '1';
      }
      else{
        s2[idx[i]] = '1';
      }
      d++;
    }
  }
  string ans;
  for(int i = 0;i < n;i++)
    ans.push_back('0');
  if(query(s1)){
    string res = solve(s1);
    for(int i = 0;i < n;i++){
      if(res[i] == '1') ans[i] = '1';
    }
  }
  if(query(s2)){
    string res = solve(s2);
    for(int i = 0;i < n;i++){
      if(res[i] == '1') ans[i] = '1';
    }
  }
  return ans;
}

int main(){
  srand(time(0));
  double p;
  cin >> n >> p >> t;
  int tam = 1;
  while(t--){
    string s;
    for(int i = 0;i < n;i++){
      s.push_back('1');
    }
    for(int i = 0;i < n;i++){
      idx.push_back(i);
    }
    random_shuffle(idx.begin(), idx.end());
    if(!query(s)){
      cout << "A ";
      for(int i = 0;i < n;i++){
        cout << 0;
      }
      cout << '\n';
      cout << flush;
      char c;
      cin >> c;
      continue;
    }
    string res = solve(s);
    idx.clear();
    cout << "A " << res << '\n';
    cout << flush;
    char c;
    cin >> c;
    if(c == 'W')
      exit(0);
  }
}

Compilation message (stderr)

Main.cpp: In function 'std::string solve(std::string)':
Main.cpp:23:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |   for(int i = 0;i < s.size();i++){
      |                 ~~^~~~~~~~~~
Main.cpp:30:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |   for(int i = 0;i < s.size();i++){
      |                 ~~^~~~~~~~~~
Main.cpp: In function 'int main()':
Main.cpp:63:7: warning: unused variable 'tam' [-Wunused-variable]
   63 |   int tam = 1;
      |       ^~~

• Subtask 0: 0 / 0
• Subtask 1: 10 / 10
• Subtask 2: 0 / 90