Submission #858148

#TimeUsernameProblemLanguageResultExecution timeMemory
858148samek08Prisoner Challenge (IOI22_prison)C++17
80 / 100
14 ms1640 KiB
    #include "prison.h"
        #include<bits/stdc++.h>
        using namespace std;
        typedef long long ll;
        typedef long double ld;
        #define rep(a,b) for (int a = 0; a < (b); ++a)
        #define pb push_back
        #define all(t) t.begin(), t.end()
         
        struct Przedzial
        {
          int l,p, idx;
        };
         
        const int MAXM = 22;
        vector<vector<int>> wyn;
        vector<bool> doit;
         
        inline Przedzial zejdz(int l, int p, int idx)
        {
          if(l == p)
          {
            return{l,p,1};
          }
          if (l+1 == p)
          {
            if(idx == l) return {l,l,1};
            return {p,p,2};
          }
          int s = (p-l+1) / 3;
          if((p-l+1) % 3 == 0 or (p-l+1) % 3 == 1)
          {
            if(idx <= l+s-1) return {l,l+s-1,1};
            else if(idx <= l+2*s-1) return {l+s,l+2*s-1,2};
            else return {l+2*s,p,3};
          }
          else
          {
            if(idx <= l+s-1) return {l,l+s-1,1};
            else if(idx <= l+2*s) return {l+s,l+2*s,2};
            else return {l+2*s+1,p,3};
          }
        }
         
        vector<vector<int>> devise_strategy(int N)
        {
          wyn.assign(MAXM+1,{});
          rep(i,MAXM+1) wyn[i].assign(N+1,-1);
          doit.assign(MAXM+1,false);
         
          wyn[0][0] = 0;
          for(int i = 1; i <= N; ++i)
          {
            if(i == 1) wyn[0][i] = -1;
            else if (i == N) wyn[0][i] = -2;
            else
            {
              wyn[0][i] = zejdz(2,N-1,i).idx;
              doit[wyn[0][i]] = true;
            }
          }
         
          for(int i = 1; i <= MAXM; ++i)
          {
            int ile = (i-1) / 3 + 1, gdzie_zeszlem = i % 3;
            if(gdzie_zeszlem == 0) gdzie_zeszlem = 3;
            bool czy_pochodze_z_A = true;
            if(ile % 2 == 0) czy_pochodze_z_A = false;
            if(czy_pochodze_z_A) wyn[i][0] = 1;
            else wyn[i][0] = 0;
            if(doit[i] == false) continue;
            for(int j = 1; j <= N; ++j)
            {
              Przedzial akt = {1,N};
              rep(k,ile-1) akt = zejdz(akt.l+1,akt.p-1,j);
              if(akt.l == j)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -2;
                else wyn[i][j] = -1;
                continue;
              }
              else if (akt.p == j)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -1;
                else wyn[i][j] = -2;
                continue;
              }
              akt = zejdz(akt.l+1,akt.p-1,j);
              if(akt.idx < gdzie_zeszlem)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -2;
                else wyn[i][j] = -1;
              }
              else if (akt.idx > gdzie_zeszlem)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -1;
                else wyn[i][j] = -2;
              }
              else if(akt.l == j)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -2;
                else wyn[i][j] = -1;
              }
              else if (akt.p == j)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -1;
                else wyn[i][j] = -2;
              }
              else
              {
                akt = zejdz(akt.l+1,akt.p-1,j);
                wyn[i][j] = ile*3+akt.idx;
                doit[wyn[i][j]] = true;
              }
            }
          } 
          for(int i = 0; i <= MAXM; ++i)
          {
            for (int j = 0; j <= N; ++j) wyn[i][j] = min(wyn[i][j],MAXM);
          } 
          
          return wyn;
        }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...