This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <cstdio>
#include <vector>
#include "library.h"
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
void Solve(int N)
{
if(N==1){
vector<int>ans={1};
Answer(ans);
return;
}
vector<int>M(N,1);
vector<int>ans;
vector<int>unknown;
for(int i=1;i<=N;i++)unknown.push_back(i);
for(int i=0;i<N;i++){//Locating an edge, letting it be L
M[i]=0;
if(Query(M)==1){
ans.push_back(i+1);
unknown.erase(remove(unknown.begin(),unknown.end(),i+1),unknown.end());
break;
}
M[i]=1;
}
while((int)ans.size()<N){
vector<int>space=unknown,a,b;
while(space.size()>1){
int halfsize=space.size()/2;
for(int i=0;i<halfsize;i++){
a.push_back(space[i]);
}
for(int i=halfsize;i<(int)space.size();i++){
b.push_back(space[i]);
}
//assert(a.size()&&b.size());
fill(M.begin(),M.end(),0);
for(auto x:a)M[x-1]=1;
int pre = Query(M);
M[ans.back()-1]=1;
int aft = Query(M);
if(pre==aft)space=a;
else space=b;
a.clear();
b.clear();
}
ans.push_back(*(space.begin()));
unknown.erase(remove(unknown.begin(),unknown.end(),*(space.begin())),unknown.end());
}
assert(ans.size()==N);
Answer(ans);
}
Compilation message (stderr)
In file included from /usr/include/c++/10/cassert:44,
from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:33,
from library.cpp:5:
library.cpp: In function 'void Solve(int)':
library.cpp:53:19: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
53 | assert(ans.size()==N);
| ~~~~~~~~~~^~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |