제출 #1338877

#제출 시각아이디문제언어결과실행 시간메모리
1338877nagorn_ph벽 칠하기 (APIO20_paint)C++20
100 / 100
164 ms15832 KiB
#include "paint.h"
#include <bits/stdc++.h>

using namespace std;

int minimumInstructions(int n, int m, int k, std::vector<int> c, std::vector<int> a, std::vector<std::vector<int>> b) {
    vector <vector <int>> con(k);
    for (int i = 0; i < m; i++) {
        for (auto e : b[i]) {
            con[e].emplace_back(i);
        }
    }
    int dp1[2][m]; memset(dp1, 0, sizeof dp1);
    vector <bool> can(n);
    vector <int> updated(m, -1);
    for (int i = 0; i < n; i++) {
        for (auto j : con[c[i]]) {
            if (i > 0 && updated[(j - 1 + m) % m] != i - 1) dp1[(i - 1)%2][(j - 1 + m) % m] = 0;
            dp1[i%2][j] = (i == 0 ? 0 : dp1[(i - 1)%2][(j - 1 + m) % m]) + 1;
            if (dp1[i%2][j] >= m) can[i] = true;
        }
        for (auto j : con[c[i]]) {
            updated[j] = i;
        }
    }
    // for (int i = 0; i < n; i++) {
    //     if (can[i]) cout << i << " ";
    // }
    vector <int> dp(n, 1e9);
    multiset <int> ms;
    for (int i = m - 1; i < n; i++){
        if (i - m - 1 >= m - 1 && can[i - m - 1]) ms.erase(ms.find(dp[i - m - 1]));
        if (!can[i]) {
            continue;
        }
        if (i - m < 0) {
            dp[i] = 1;
            ms.emplace(dp[i]);
            continue;
        }
        if (ms.empty()) return -1;
        dp[i] = *ms.begin() + 1;
        ms.emplace(dp[i]);
    }
    return (dp[n-1] == 1e9 ? -1 : dp[n-1]);
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...