Submission #397357

#	Time	Username	Problem	Language	Result	Execution time	Memory
397357		rocks03	Painting Walls (APIO20_paint)	C++14	28 / 100	1231 ms	524292 KiB

paint

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int, int>
#define pll pair<ll, ll>
#define ff first
#define ss second
#define pb push_back
#define SZ(x) ((int)(x).size())
#define all(x) x.begin(), x.end()
#define debug(x) cout << #x << ": " << x << " "
#define nl cout << "\n"
#define rep(i, a, b) for(int i = (a); i < (b); i++)
#define per(i, a, b) for(int i = (a); i >= (b); i--)
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int minimumInstructions(int N, int M, int K, vector<int> C, vector<int> A, vector<vector<int>> B){
    vector<int> pos[K + 1];
    rep(i, 0, N) pos[C[i]].pb(i);
    vector<int> workers[N + 1];
    rep(i, 0, M){
        for(int col : B[i]){
            for(int p : pos[col]){
                workers[p].pb(i);
            }
        }
    }
    vector<bool> v(N + 1, false);
    vector<pii> work[N + 1];
    for(int x : workers[0]){
        work[0].pb({x, 0});
    }
    rep(y, 1, N + 1){
        for(auto [x, old_y] : work[y - 1]){
            int i = (x + 1) % M, l = y - old_y, j = y;
            if(l == M){
                v[old_y] = true; continue;
            }
            auto it = lower_bound(all(B[i]), C[j]);
            if(it != B[i].end() && *it == C[j]){
                work[y].pb({i, old_y});
            }
        }
        work[y - 1].clear();
        if(y + M - 1 < N){
            for(int x : workers[y]){
                work[y].pb({x, y});
            }
        }
    }
    if(!v[0]) return -1;
    vector<int> dp(N, INT_MAX);
    dp[0] = 1;
    multiset<int> s, ms[N + 1];
    rep(i, 0, N){
        for(int x : ms[i]){
            s.erase(s.find(x));
        }
        if(SZ(s)){
            dp[i] = *s.begin() + 1;
        }
        if(dp[i] == INT_MAX) return -1;
        if(v[i]){
            s.insert(dp[i]);
            int r = min(N - 1, i + M) + 1;
            ms[r].insert(dp[i]);
            if(i + M - 1 >= N - 1) return dp[i];
        }
    }
    return -1;
}

Compilation message (stderr)

paint.cpp: In function 'int minimumInstructions(int, int, int, std::vector<int>, std::vector<int>, std::vector<std::vector<int> >)':
paint.cpp:34:18: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   34 |         for(auto [x, old_y] : work[y - 1]){
      |                  ^

• Subtask 1: 0 / 12
• Subtask 2: 15 / 15
• Subtask 3: 13 / 13
• Subtask 4: 0 / 23
• Subtask 5: 0 / 37