# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1253628 | denislav | Unscrambling a Messy Bug (IOI16_messy) | C++20 | 0 ms | 0 KiB |
# include <iostream>
# include <vector>
# include <algorithm>
using namespace std;
# include "messy.h"
//# include "grader.cpp"
const int MAX=128;
int n,LOG;
string s[MAX];
void build_level(int l, int r, int level)
{
if(l==r) return ;
int mid=(l+r)/2;
for(int i=l;i<=mid;i++) s[level][i]='1';
build_level(l,mid,level+1);
build_level(mid+1,r,level+1);
}
int find_level(int x, int y, int l=0, int r=n-1, int level=0)
{
int mid=(l+r)/2;
if(x<=mid and mid<y) return level;
if(y<=mid) return find_level(x,y,l,mid,level+1);
else return find_level(x,y,mid+1,r,level+1);
}
std::vector<int> restore_permutation(int N, int w, int r)
{
n=N;
LOG=31-__builtin_clz(n);
for(int i=0;i<LOG;i++) s[i]=string(n,'0');
build_level(0,n-1,0);
for(int i=0;i<LOG;i++) add_element(s[i]);
compile_set();
for(int x=0;x<n;x++)
{
for(int y=x+1;y<n;y++)
{
int level=find_level(x,y);
string curr=s[level];
swap(curr[x],curr[y]);
if(check_element(curr))
{
vector<int> ans(n);
iota(ans.begin(),ans.end(),0);
swap(ans[x],ans[y]);
return ans;
}
}
}
vector<int> ans(n);
iota(ans.begin(),ans.end(),0);
return ans;
}
/*
4 128 128
0 1 2 3
*/