제출 #850123

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
850123		tvladm2009	벽 칠하기 (APIO20_paint)	C++17	12 / 100	35 ms	12820 KiB

paint

#include "paint.h"

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

int n, m, k;

int minimumInstructions(int N, int M, int K, vector<int> C, vector<int> A, vector<vector<int>> B) {
        vector<int> a, c;
        vector<vector<int>> b;

        n = N;
        m = M;
        k = K;
        c = C;
        a = A;
        b = B;
        assert((int)c.size() == n);
        assert((int)a.size() == m);
        assert((int)b.size() == m);
        for (int i = 0; i < m; i++) {
                assert((int)b[i].size() == a[i]);
        }

        vector<int> f(k, 0);
        for (int i = 0; i < m; i++) {
                for (auto& it : b[i]) {
                        f[it]++;
                }
        }
        vector<int> has(k, -1);
        vector<int> need(n, 0);
        for (int i = 0; i < m; i++) {
                for (auto& it : b[i]) {
                        has[it] = i;
                }
        }
        for (int i = 0; i < n; i++) {
                if (f[c[i]] == 0) {
                        return -1;
                }
                need[i] = has[c[i]];
        }
        int sol = 0, pos = need[0], cur = 1;
        for (int i = 1; i < n; i++) {
                bool ok = false;
                ok |= (need[i] == (pos + 1) % m);

                if (!ok) {
                        if (cur < m) {
                                return -1;
                        }
                        sol += (cur + m - 1) / m;
                        cur = 0;
                }
                cur++;
                pos = need[i];
        }
        if (cur < m) {
                return -1;
        }
        sol += (cur + m - 1) / m;
        return sol;
}

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