# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
603961 | neki | Unscrambling a Messy Bug (IOI16_messy) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "messy.h"
#define vc vector
using namespace std;
void badd(int l, int r, int n){
if(l+1==r) return;
int mid=(l+r)/2;
string s(n, '1');
for(int i=l; i<r;++i) s[i]='0';
for(int i=l; i<mid;++i){
string t=s;
t[i]='1';
add(t);
}
badd(l, mid, n);
badd(mid, r, n);
}
void bquery(int l, int r, int n, vc<int>& ans){
if(l+1==r) return;
int mid=(l+r)/2;
string s(n, '1');
for(int i=l; i<r;++i) s[ans[i]]='0';
set<int> ls, rs;
for(int i=l; i<r;++i){
string t=s;
t[ans[i]]='1';
if(check_element(t)) ls.insert(ans[i]);
else rs.insert(ans[i]);
}
vc<int> lv(ls.begin(), ls.end()), rv(rs.begin(), rs.end());
assert(lv.size()==(mid-l) and rv.size()==(r-mid));
for(int i=l;i<mid;++i) ans[i]=lv[i-l];
for(int i=mid;i<r;++i) ans[i]=rv[i-mid];
bquery(l, mid, n, ans);
bquery(mid, r, n, ans);
}
vc<int> inv(vc<int> ans){
vc<int> ret(ans.size());
for(int i=0;i<ans.size();++i) ret[ans[i]]=i;
return ret;
}
vc<int> restore_permutation(int n, int w, int r){
badd(0, n, n);
compileset();
vc<int> ans(n); for(int i=0;i<n;++i) ans[i]=i;
bquery(0, n, n, ans);
return inv(ans);
}