# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
301473 | matheo_apd | Unscrambling a Messy Bug (IOI16_messy) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "messy.h"
using namespace std;
#define all(x) begin(x), end(x)
#define pb push_back
#define maxn 130
int n_;
int ans[maxn];
void add(int l, int r){
if(l == r) return;
string c;
for(int i = 0; i < n_; i++){
if(l <= i and i <= r) c.pb('1');
else c.pb('0');
}
int mid = (l+r)/2;
for(int i = l; i <= mid; i++){
c[i] = '0';
add_element(c);
c[i] = 1;
}
add(l, mid); add(mid+1, r);
}
void solve(int l, int r, vector<int> caras){
if(l == r){
ans[caras[0]] = l;
return;
}
vector<int> L, R;
string c;
for(int i = 0; i < n_; i++){
if(binary_search(all(caras), i)) c.pb('1');
else c.pb('0');
}
for(auto i : caras){
c[i] = '0';
if(check_element(c)) L.push_back(i);
else R.push_back(i);
c[i] = '1';
}
int mid = (l+r)/2;
solve(l, mid, L);
solve(mid+1, r, R);
}
int[] restore_permutation(int n, int w, int r){
n_ = n;
add(0, n-1);
compile_set();
vector<int> caras;
for(int i = 0; i < n; i++)
caras.push_back(i);
solve(0, n-1, caras);
int resp[n];
for(int i = 0; i < n; i++)
resp[i] = ans[i];
return resp;
}