Submission #1351440

#TimeUsernameProblemLanguageResultExecution timeMemory
1351440JohanCarnival (CEOI14_carnival)C++20
100 / 100
6 ms504 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int MOD = 1e9 + 7;
const int INF = 1e18;
const int N = 1e3 + 5;
vector < int > arr = {0, 1, 2, 1, 1, 1};
int ask(vector < int > x){
  int n = x.size();
  cout << n << ' ';
  for(int i : x)
    cout << i << ' ';
  cout << endl;
  int rs = 0;
  // set < int > st;
  // for(int i : x)
  //   st.insert(arr[i]);
  // rs = (int)st.size();
  cin >> rs;
  return rs;
}
vector < int > cut(int l, int r){
  vector < int > x;
  for(int i = l; i <= r; i++)
    x.push_back(i);
  return x;
}
signed main(){
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  int n;
  cin >> n;
  int clr = 1;
  vector < int > ans(n + 1, 0);
  for(int i = 1; i <= n; i++){
    if(ans[i] == 0)
      ans[i] = clr++;
    int l = i + 1, r = n;
    int best = -1;
    while(r >= l){
      int mid = (l + r) >> 1;
      vector < int > cur = cut(i + 1, mid);
      vector < int > curc = cur;
      curc.push_back(i);
      if(ask(cur) == ask(curc)){
        best = mid;
        r = mid - 1;
      }
      else l = mid + 1;
    }
    if(best == -1)
      continue;
    ans[best] = ans[i];
  }
  cout << "0 ";
  for(int i = 1; i <= n; i++)
    cout << ans[i] << ' ';
  cout << endl;
}
/*

*/
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...