Submission #297976

#	Time	Username	Problem	Language	Result	Execution time	Memory
297976		Autoratch	Ancient Books (IOI17_books)	C++14	12 / 100	975 ms	1064 KiB

books

#include "books.h"
#include <bits/stdc++.h>
using namespace std;
#define pii pair<int,tuple<int,int,int> > 

const int N = 2000;

int n;
map<int,vector<int> > ma;
map<vector<int>,int> mb;
int res[N][5][5];
priority_queue<pii,vector<pii>,greater<pii> > q;
bool visited[N][5][5];

long long minimum_walk(std::vector<int> p, int s) 
{
    n = p.size();
    vector<int> v;
    for(int i = 0;i < n;i++) v.push_back(i);
    int cnt = 0;
    if(n==4) for(int i = 0;i <= 4;i++) for(int j = 0;j <= 4;j++) for(int k = 0;k <= 4;k++) for(int l = 0;l <= 4;l++) 
    {
        vector<int> v = {i,j,k,l};
        ma[cnt] = v;
        mb[v] = cnt;
        cnt++;
    }
    else if(n==3) for(int i = 0;i <= 3;i++) for(int j = 0;j <= 3;j++) for(int k = 0;k <= 3;k++)
    {
        vector<int> v = {i,j,k};
        ma[cnt] = v;
        mb[v] = cnt;
        cnt++;
 
    }
    else if(n==2) for(int i = 0;i <= 2;i++) for(int j = 0;j <= 2;j++)
    {
        vector<int> v = {i,j};
        ma[cnt] = v;
        mb[v] = cnt;
        cnt++;
    }
    else for(int i = 0;i <= 1;i++)
    {
        vector<int> v = {i};
        ma[cnt] = v;
        mb[v] = cnt;
        cnt++;
    }   
    for(int i = 0;i < N;i++) for(int j = 0;j <= n;j++) for(int k = 0;k <= n;k++) res[i][j][k] = INT_MAX;
    res[mb[p]][s][n] = 0;
    q.push({0,{mb[p],s,n}});
    while(!q.empty())
    {
        auto [per,id,hn] = q.top().second;
        q.pop();
        if(visited[per][id][hn]) continue;
        visited[per][id][hn] = true;
        vector<int> now = ma[per];
        for(int i = 0;i < n;i++) if(res[per][id][hn]+abs(id-i)<res[per][i][hn])
        {
            res[per][i][hn] = res[per][id][hn]+abs(id-i);
            q.push({res[per][i][hn],{per,i,hn}});
        }
        vector<int> nx = now;
        int hnd = nx[id];
        nx[id] = hn;
        if(res[per][id][hn]<res[mb[nx]][id][hnd])
        {
            res[mb[nx]][id][hnd] = res[per][id][hn];
            q.push({res[mb[nx]][id][hnd],{mb[nx],id,hnd}});
        }
    }
    vector<int> ans;
    for(int i = 0;i < n;i++) ans.push_back(i);
    return res[mb[ans]][0][n];
}

Compilation message (stderr)

books.cpp: In function 'long long int minimum_walk(std::vector<int>, int)':
books.cpp:55:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   55 |         auto [per,id,hn] = q.top().second;
      |              ^

• Subtask 1: 12 / 12
• Subtask 2: 0 / 10
• Subtask 3: 0 / 28
• Subtask 4: 0 / 20
• Subtask 5: 0 / 30