Submission #691828

#	Time	Username	Problem	Language	Result	Execution time	Memory
691828		PoonYaPat	Prisoner Challenge (IOI22_prison)	C++17	0 / 100	6 ms	1108 KiB

prison

#include "prison.h"
#include <bits/stdc++.h>
using namespace std;

vector<int> v[5001];

void trans(int x, int l, int r) {
    if (x==l) v[x].push_back(-1);
    else if (x==r) v[x].push_back(-2);
    else {
        int par=(r-l-1)/3;
        if (x<=l+par) {
            v[x].push_back(1);
            trans(x,l+1,l+par);
        } else if (x<=l+2*par) {
            v[x].push_back(2);
            trans(x,l+par+1,l+2*par);
        } else {
            v[x].push_back(3);
            trans(x,l+2*par+1,r-1);
        }
    }
}

int s[5001][5001];
vector<int> a={3,3,3,3,3,2,2,1};

vector<vector<int>> devise_strategy(int n) {
    for (int i=1; i<=n; ++i) trans(i,1,5588);

    s[0][0]=1; //look at B
    for (int i=1; i<=n; ++i) {
        if (v[i][0]==-1) s[0][i]=-2;
        else if (v[i][0]==-2) s[0][i]=-1;
        else s[0][i]=v[i][0];
    }

    int idx=0;
    for (int k=0; k<a.size(); ++k) {
        for (int i=idx+1; i<=idx+a[k]; ++i) {

            s[i][0]=k%2;

            //read i -> range i-idx
            for (int j=1; j<=n; ++j) {
                //open bag then see j

                if (v[j].size()<k+1) s[i][j]=0;
                else if (v[j][k]==-1 || v[j][k]<i-idx) {
                    if (k%2==0) s[i][j]=-1;
                    else s[i][j]=-2;

                } else if (v[j][k]==-2 || v[j][k]>i-idx) {
                    if (k%2==0) s[i][j]=-2;
                    else s[i][j]=-1;

                } else { //v[j][k]==i-idx

                    if (v[j][k+1]==-1) {
                        if (k%2==0) s[i][j]=-1;
                        else s[i][j]=-2;

                    } else if (v[j][k+1]==-2) {
                        if (k%2==0) s[i][j]=-2;
                        else s[i][j]=-1;

                    } else s[i][j]=v[j][k+1]+idx+a[k];

                }
            }

        }
        idx+=a[k];
    }

    vector<vector<int>> ans;
    for (int i=0; i<=20; ++i) {
        vector<int> temp;
        for (int j=0; j<=n; ++j) temp.push_back(s[i][j]);
        ans.push_back(temp);
    }
    return ans;
}

Compilation message (stderr)

prison.cpp: In function 'std::vector<std::vector<int> > devise_strategy(int)':
prison.cpp:39:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |     for (int k=0; k<a.size(); ++k) {
      |                   ~^~~~~~~~~
prison.cpp:48:32: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   48 |                 if (v[j].size()<k+1) s[i][j]=0;
      |                     ~~~~~~~~~~~^~~~

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