# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
594906 | TimDee | Mechanical Doll (IOI18_doll) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
void create_circuit(int m, vector<int>a) {
int n=a.size();
vector<int> cnt(m+1,0);
int mx=0;
for (int i=0; i<n; ++i) {
cnt[a[i]]++;
mx=max(mx,cnt[a[i]]);
}
if (mx==1) {
vector<int> c(m+1,0);
c[0]=a[0];
for (int i=1; i<n; ++i) {
c[a[i-1]]=a[i];
}
vector<int> x(0), y(0);
answer(c,x,y);
}/* else if (mx==2) {
} else if (mx<=4) {
} */else if (m==1) {
int size=1;
while (4*size<n) size<<=1;
vector<int> x(2*size+1), y(2*size+1), state(2*size+1,0), c = {1, -(2*size)};
for (int i=1; i<=size-1; ++i) {
x[i-1] = -(2*i);
y[i-1] = -(2*i+1);
}
x[2*size-1]=-(2*size+1);
y[2*size-1]=-1;
x[2*size] = y[2*size] = 1;
int cnt=0, p=0;
while (cnt<2*size) {
//cout<<p<<"->";
if (p>=size-1) {
//cout<<"added "<<p<<' '<<cnt<<'\n';
if (state[p]) {
if (cnt<n-1-2*size) y[p]=1;
else y[p]=-(2*size);
} else {
if (cnt<n-1-2*size) x[p]=1;
else x[p]=-(2*size);
}
state[p]^=1;
++cnt;
p=0;
continue;
}
if (!state[p]) {
state[p]^=1;
p<<=1; ++p;
} else {
state[p]^=1;
++p; p<<=1;
}
}
y[2*size-2]=0;
answer(c,x,y);
} else {
vector<int> c(m+1),x,y;
c[0]=a[0];
vector<tree> v(m+1);
for (int i=1; i<=m; ++i) v[i].build(cnt[i],cnt[i]);
int in=0, p=0;
while (in<n) {
int ind = a[in];
while (p<v[ind].size-1) {
if (!v[ind].state[p]) {
v[ind].state[p]^=1;
p<<=1;
++p;
} else {
v[ind].state[p]^=1;
++p;
p<<=1;
}
}
if (!v[ind].state[p]) {
v[ind].x[p]= ++in<n ? a[in] : 0;
v[ind].state[p]^=1;
} else {
v[ind].y[p]= ++in<n ? a[in] : 0;
v[ind].state[p]^=1;
}
}
for(int i=1; i<=m; ++i) {
int sz=x.size();
for (auto X:v[i].x) x.push_back( X>=0 ? X : (X-sz) );
for (auto Y:v[i].y) y.push_back( Y>=0 ? Y : (Y-sz) );
c[i]=-1-sz;
}
answer(c,x,y);
}
}