Submission #397345

#	Time	Username	Problem	Language	Result	Execution time	Memory
397345		rocks03	Painting Walls (APIO20_paint)	C++14	0 / 100	1 ms	204 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);
    rep(y, 0, N - M + 1){
        for(int x : workers[y]){
            bool ok = true;
            rep(l, 0, M){
                int i = (x + l) % M, j = y + l;
                auto it = lower_bound(all(B[i]), C[j]);
                if(it != B[i].end() && *it == C[j]) continue;
                ok = false; break;
            }
            if(ok){
                v[y] = true; break;
            }
        }
    }
    if(!v[0]) return -1;
    vector<int> dp(N, INT_MAX);
    dp[0] = 0;
    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] = min(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(dp[N - 1] == INT_MAX) dp[N - 1] = -1;
    return dp[N - 1];
}

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