제출 #765833

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
765833		Khizri	죄수들의 도전 (IOI22_prison)	C++17	51.50 / 100	37 ms	1340 KiB

prison

#include "prison.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define F first
#define S second
#define INF 1e18
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define pii pair<int,int>
#define pll pair<ll,ll>
#define OK cout<<"Ok"<<endl;
#define MOD (ll)(1e9+7)
const int mxn=5e3+5;
int digit(int k,int idx){
    vector<int>vt;
    while(k>0){
        vt.pb(k%3);
        k/=3;
    }
    while(vt.size()<8){
        vt.pb(0);
    }
    return vt[7-idx];
}
vector<vector<int>> devise_strategy(int n) {
    vector<vector<int>>ans;
    for(int id=0;id<30;id++){
        if(id%4==0){
            vector<int>vt;
            vt.pb(1);
            for(int i=1;i<=n;i++){
                int q=digit(i,id/4);
                if(q==0){
                    if(id!=28){
                        vt.pb(id+1);
                    }
                    else{
                        vt.pb(-2);
                    }
                }
                else if(q==1){
                    if(id!=28){
                        vt.pb(id+2);
                    }
                    else{
                        vt.pb(id+1);
                    }
                }
                else{
                    if(id!=28){
                        vt.pb(id+3);
                    }
                    else{
                        vt.pb(-1);
                    }
                }
            }
            ans.pb(vt);
        }
        else{
            vector<int>vt;
            vt.pb(0);
            int nxt=(id/4+1)*4;
            int k=id%4-1;
            for(int i=1;i<=n;i++){
                int q=digit(i,id/4);
                if(id==29){
                    if(q==0){
                        vt.pb(-1);
                    }
                    else{
                        vt.pb(-2);
                    }
                }
                else{
                    if(q<k){
                        vt.pb(-1);
                    }
                    else if(q>k){
                        vt.pb(-2);
                    }
                    else{
                        if(id<28){
                            vt.pb(nxt);
                        }
                        else{
                            vt.pb(0);
                        }
                    }
                }
            }
            ans.pb(vt);
        }
    }
    /*
    for(int i=0;i<ans.size();i++){
        for(int v:ans[i]){
            cout<<v<<' ';
        }
        cout<<endl;
    }
    */
    return ans;
}

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