Submission #858155

#	Time	Username	Problem	Language	Result	Execution time	Memory
858155		samek08	Prisoner Challenge (IOI22_prison)	C++17	90 / 100	10 ms	1372 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 = 21;
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;
}

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