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 <bits/stdc++.h>
#include "xylophone.h"
using namespace std;
void solve(int N) {
int a[N+1];
memset(a, 0, sizeof(a));
set<int> s;
// binary search pos of 1
int lower = 1, upper = N;
while(upper-lower>1){
int mid = (upper+lower)/2;
int x = query(mid, N);
if(x==N-1) lower = mid;
else upper = mid;
}
int pos1 = lower;
a[pos1] = 1;
s.insert(1);
if(pos1<N){
// go to right
int prev = query(pos1, pos1+1);
a[pos1+1] = 1 + prev;
s.insert(a[pos1+1]);
for (int i=pos1+1; i<N; i++){
int next = query(i, i+1);
if(a[i]+next>N) a[i+1] = a[i]-next;
else if(a[i]-next<1) a[i+1] = a[i]+next;
else if(s.find(a[i]+next)!=s.end()) a[i+1] = a[i]-next;
else if(s.find(a[i]-next)!=s.end()) a[i+1] = a[i]+next;
else{
int cur = query(i-1, i+1);
if(cur==prev+next){
if(a[i]>a[i-1]) a[i+1] = a[i]+next;
else a[i+1] = a[i]-next;
}
else{
if(a[i]>a[i-1]) a[i+1] = a[i]-next;
else a[i+1] = a[i]+next;
}
}
prev = next;
}
}
if(pos1>1){
// go to left
int prev = query(pos1-1, pos1);
a[pos1-1] = 1 + prev;
s.insert(a[pos1-1]);
for (int i=pos1-1; i>1; i--){
int next = query(i-1, i);
if(a[i]+next>N) a[i-1] = a[i]-next;
else if(a[i]-next<1) a[i-1] = a[i]+next;
else if(s.find(a[i]+next)!=s.end()) a[i-1] = a[i]-next;
else if(s.find(a[i]-next)!=s.end()) a[i-1] = a[i]+next;
else{
int cur = query(i-1, i+1);
if(cur==prev+next){
if(a[i]>a[i+1]) a[i-1] = a[i]+next;
else a[i-1] = a[i]-next;
}
else{
if(a[i]>a[i+1]) a[i-1] = a[i]-next;
else a[i-1] = a[i]+next;
}
}
prev = next;
}
}
for (int i=1; i<=N; i++) answer(i, a[i]);
return;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |