#include "prison.h"
#include <vector>
#include <iostream>
using namespace std;
#define all(x) (X).begin(), (x).end()
vector<int> mb(int N){
  int stn = 3*3*3*3*3*3*3;
  vector<int> repr;
  for(int num = stn;num>=1;num/=3){
    int cc = 0;
    while(num<=N){
      N-=num;
      cc++;
    }
    repr.push_back(cc);
  }
  return repr;
}
vector<vector<int>> devise_strategy(int N) {
  
  vector<vector<int>> s(25, vector<int>(N+1));
  for(int i=0;i<25;i++){
    if(i==0){
      s[i][0]=0;
      for(int j = 1;j<=N;j++){
        auto b3 = mb(j);
        s[i][j]=i+b3[0]+1;
      }
      continue;
    }
    int bnum=(i-1)%3;
    int nbid = ((i-1)/3);
    if(((i-1)/3)%2==0){
      s[i][0]=1;
      for(int j = 1;j<=N;j++){
        auto b3 = mb(j);
        if (b3[nbid]<bnum)s[i][j]=-2;
        else if (b3[nbid]>bnum)s[i][j]=-1;
        else if (i<22) s[i][j]=i+b3[nbid+1]+1;
        else s[i][j]=-1;
      }
    }
    else{
      s[i][0]=0;
      for(int j = 1;j<=N;j++){
        auto b3 = mb(j);
        if (b3[nbid]<bnum)s[i][j]=-1;
        else if (b3[nbid]>bnum)s[i][j]=-2;
        else if (i<22) s[i][j]=i+b3[nbid+1]+1;
        else s[i][j] = -1;
      }
    }
  }
  return s;
}
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... |