Submission #776012

# Submission time Handle Problem Language Result Execution time Memory
776012 2023-07-07T08:31:03 Z I_Love_EliskaM_ Mechanical Doll (IOI18_doll) C++14
100 / 100
345 ms 38772 KB
#include "doll.h"
#include <bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i=0;i<n;++i)
#define ll long long
#define pb push_back
#define all(x) x.begin(), x.end()
const ll inf = 1e18;
#define pi pair<int,int>
#define f first
#define s second

void mysim(vector<int> c, vector<int> x, vector<int> y) {
  int p=0;
  int z=1;
  int k=x.size();
  vector<int> s(k,0);
  while (z||p) {
    if (!p) {
      z=0; p=c[p];
    } else if (p>0) {
      cout<<p<<"->\n";
      p=c[p];
    } else {
      cout<<"? "<<p<<'\n';
      int i=-p-1;
      if (s[i]==0) p=x[i];
      else p=y[i];
      s[i]^=1;
    }
  }
}

void create_circuit(int m, vector<int>a) {
  int n=a.size();
  map<int,int> cnt;
  int mx=0;
  for(auto&x:a) {
    cnt[x]++;
    mx=max(mx,cnt[x]);
  }
  if (mx<=4) {
    vector<int> c(m+1,0);
    c[0]=a[0];
    vector<int> x,y;
    int nxt=0;
    for(auto&it:cnt) {
      int v=it.f, k=it.s;
      if (k==1) continue;
      if (k==2) {
        x.pb(0), y.pb(0);
        c[v]=-1-nxt++;
      } else {
        x.pb(-1-(nxt+1)), y.pb(-1-(nxt+2));
        forn(i,2) x.pb(0), y.pb(0);
        if (k==3) y[nxt+1]=-1-(nxt);
        c[v]=-nxt-1;
        nxt+=3;
      }
    }
    map<int,int> ok;
    a.pb(0);
    for(int i=0;i<n;++i) {
      int k=cnt[a[i]];
      if (k==1) {
        c[a[i]]=a[i+1];
      } else if (k==2) {
        if (ok[a[i]]) {
          y[-c[a[i]]-1]=a[i+1];
        } else {
          ++ok[a[i]];
          x[-c[a[i]]-1]=a[i+1];
        }
      } else if (k==3) {
        if (ok[a[i]]==0) {
          x[-1-x[-c[a[i]]-1]]=a[i+1];
        } else if (ok[a[i]]==1) {
          x[-1-y[-c[a[i]]-1]]=a[i+1];
        } else {
          y[-1-y[-c[a[i]]-1]]=a[i+1];
        }
        ++ok[a[i]];
      } else {
        if (ok[a[i]]==0) {
          x[-1-x[-c[a[i]]-1]]=a[i+1];
        } else if (ok[a[i]]==1) {
          x[-1-y[-c[a[i]]-1]]=a[i+1];
        } else if (ok[a[i]]==2) {
          y[-1-x[-c[a[i]]-1]]=a[i+1];
        } else {
          y[-1-y[-c[a[i]]-1]]=a[i+1];
        }
        ++ok[a[i]];
      }
    }
    //answer(c,x,y);
    //return;
  }

  vector<int> c(m+1,-1); c[0]=a[0];
  int sz=1; while (2*sz<n) sz<<=1;
  vector<int> x(2*sz-1,-1), y(2*sz-1,-1);
  forn(i,sz-1) x[i]=2*i+1, y[i]=2*i+2;
  forn(i,sz-1) x[i]=-1-x[i], y[i]=-1-y[i];
  vector<int> s(2*sz-1,1);
 
  vector<pi> z;
  forn(it,2*sz) {
    int p=0;
    while (p<sz-1) {
      s[p]^=1;
      p=2*p+1+s[p];
    }
    s[p]^=1;
    z.pb({p,s[p]});
  }

  if (n&1) {
    c[0]=-1;
    a.pb(0);
    ++n;
  } else {
    forn(i,n-1) a[i]=a[i+1];
    a[n-1]=0;
  }

  vector<int> ok(2*sz);
  forn(i,sz-n/2) ok[sz-1+i]=1;

  int p=0;
  for(auto&it:z) {
    if (ok[it.f]) continue;
    if (it.s) y[it.f]=a[p];
    else x[it.f]=a[p];
    ++p;
  }
  //forn(i,2*sz-1) {
  //  cout<<x[i]<<' '<<y[i]<<'\n';
  //}
  //return;
  
  vector<int> fs,sc;
  set<int> st; forn(i,2*sz-1) st.insert(i);
  for(int i=2*sz-2; i>=0; --i) {
    if (x[i]==-1 && y[i]==-1) {
      st.erase(i);
      if (i&1) {
        x[(i-1)/2]=-1;
      } else {
        y[(i-1)/2]=-1;
      }
    }
  }
  forn(i,2*sz-1) {
    if (!st.count(i)) continue;
    fs.pb(x[i]);
    sc.pb(y[i]);
  }

  map<int,int> comp; int mex=0;
  for(auto&x:st) comp[x]=mex++;
  forn(i,sz-1) {
    if (fs[i]<0) fs[i]=-1-comp[-fs[i]-1];
    if (sc[i]<0) sc[i]=-1-comp[-sc[i]-1];
  }
  //forn(i,fs.size()) {
  //  cout<<fs[i]<<' '<<sc[i]<<'\n';
  //}
  //mysim(c,fs,sc);
  //return;
 
  answer(c,fs,sc);
}
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 84 ms 15000 KB Output is correct
3 Correct 101 ms 16336 KB Output is correct
4 Correct 0 ms 300 KB Output is correct
5 Correct 7 ms 1740 KB Output is correct
6 Correct 139 ms 22444 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 84 ms 15000 KB Output is correct
3 Correct 101 ms 16336 KB Output is correct
4 Correct 0 ms 300 KB Output is correct
5 Correct 7 ms 1740 KB Output is correct
6 Correct 139 ms 22444 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 198 ms 24896 KB Output is correct
9 Correct 231 ms 30072 KB Output is correct
10 Correct 339 ms 38772 KB Output is correct
11 Correct 1 ms 212 KB Output is correct
12 Correct 1 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 84 ms 15000 KB Output is correct
3 Correct 101 ms 16336 KB Output is correct
4 Correct 0 ms 300 KB Output is correct
5 Correct 7 ms 1740 KB Output is correct
6 Correct 139 ms 22444 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 198 ms 24896 KB Output is correct
9 Correct 231 ms 30072 KB Output is correct
10 Correct 339 ms 38772 KB Output is correct
11 Correct 1 ms 212 KB Output is correct
12 Correct 1 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
14 Correct 345 ms 36920 KB Output is correct
15 Correct 243 ms 26768 KB Output is correct
16 Correct 330 ms 35840 KB Output is correct
17 Correct 1 ms 212 KB Output is correct
18 Correct 0 ms 300 KB Output is correct
19 Correct 1 ms 212 KB Output is correct
20 Correct 341 ms 37660 KB Output is correct
21 Correct 1 ms 212 KB Output is correct
22 Correct 1 ms 212 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 308 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 1 ms 300 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 304 KB Output is correct
2 Correct 116 ms 20720 KB Output is correct
3 Correct 139 ms 24836 KB Output is correct
4 Correct 180 ms 33052 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 304 KB Output is correct
2 Correct 116 ms 20720 KB Output is correct
3 Correct 139 ms 24836 KB Output is correct
4 Correct 180 ms 33052 KB Output is correct
5 Correct 223 ms 35260 KB Output is correct
6 Correct 215 ms 34236 KB Output is correct
7 Correct 217 ms 34592 KB Output is correct
8 Correct 208 ms 33588 KB Output is correct
9 Correct 140 ms 24912 KB Output is correct
10 Correct 187 ms 33348 KB Output is correct
11 Correct 188 ms 33100 KB Output is correct
12 Correct 153 ms 24820 KB Output is correct
13 Correct 124 ms 21496 KB Output is correct
14 Correct 164 ms 25840 KB Output is correct
15 Correct 173 ms 26304 KB Output is correct
16 Correct 4 ms 1108 KB Output is correct
17 Correct 112 ms 20676 KB Output is correct
18 Correct 104 ms 20728 KB Output is correct
19 Correct 151 ms 24812 KB Output is correct
20 Correct 197 ms 33152 KB Output is correct
21 Correct 200 ms 33052 KB Output is correct
22 Correct 187 ms 32956 KB Output is correct