Submission #858042

#	Time	Username	Problem	Language	Result	Execution time	Memory
858042		samek08	Prisoner Challenge (IOI22_prison)	C++17	80 / 100	11 ms	1628 KiB

prison

#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(1,N,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,akt.p,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,akt.p,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,akt.p,j);
            if(akt.l == akt.p)
            {
              if(akt.idx == 1)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -2;
                else wyn[i][j] = -1;
              }
              else if (akt.idx == 3)
              {
                if(czy_pochodze_z_A) wyn[i][j] = -1;
                else wyn[i][j] = -2;
              }
              else
              {
                wyn[i][j] = ile*3+akt.idx;
                doit[wyn[i][j]] = true;
              }
            }
            else
            {
              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;
    }

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 70 / 90