Submission #697514

#	Time	Username	Problem	Language	Result	Execution time	Memory
697514		whqkrtk04	Painting Walls (APIO20_paint)	C++17	100 / 100	1004 ms	261976 KiB

paint

#include "paint.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<int, pii> piii;
typedef pair<ll, ll> pll;
typedef pair<ll, pll> plll;
#define fi first
#define se second
const int INF = 1e9+1;
const int P = 1000000007;
const ll LLINF = (ll)1e18+1;
template <typename T1, typename T2>
ostream& operator<<(ostream& os, const pair<T1, T2>& p) { os << p.fi << " " << p.se; return os; }
template <typename T>
ostream& operator<<(ostream& os, const deque<T>& v) { for(auto i : v) os << i << " "; os << "\n"; return os; }
template <typename T>
ostream& operator<<(ostream& os, const vector<T>& v) { for(auto i : v) os << i << " "; os << "\n"; return os; }

int minimumInstructions(int N, int M, int K, vector<int> C, vector<int> A, vector<vector<int>> B) {
    vector<vector<int>> work(K);
    for(int i = 0; i < M; i++)
        for(auto j : B[i])
            work[j].push_back(i);
    //cout << " " << work;
    vector<vector<int>> can(M);
    for(int i = 0, idx = 0; i < N; i++, idx = (idx-1+M)%M)
        for(auto j : work[C[i]])
            can[(j+idx)%M].push_back(i);
    //cout << " " << can;
    vector<bool> paint(N, false);
    for(int i = 0; i < M; i++)
        for(int j = M-1, k = 0; j < can[i].size(); j++, k++) 
            if(can[i][j]-can[i][k] == M-1)
                paint[can[i][j]] = true;
    //cout << paint;
    vector<int> dp(N, INF);
    if(paint[M-1]) fill(dp.begin(), dp.begin()+M, 1);
    for(int i = M; i < N; i++) {
        if(!paint[i]) continue;
        dp[i] = min(dp[i], dp[i-M]+1);
        for(int j = i-1; j >= 0 && !paint[j]; j--) dp[j] = dp[i];
    }
    if(dp.back() == INF) return -1;
    else return dp.back();
}

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:35: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |         for(int j = M-1, k = 0; j < can[i].size(); j++, k++)
      |                                 ~~^~~~~~~~~~~~~~~

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