Submission #500928

#	Time	Username	Problem	Language	Result	Execution time	Memory
500928		sidon	Painting Walls (APIO20_paint)	C++17	100 / 100	461 ms	14804 KiB

paint

#include <bits/stdc++.h>
using namespace std;
#include "paint.h"
#define V vector

int minimumInstructions(int N, int M, int K, V<int> C, V<int> A, V<V<int>> B) {
	V<int> cand[K], x(M), y(M), dp(N+1);

	for(int i = 0; i < M; i++)
		for(int &j : B[i]) cand[j].push_back(i);

	for(int i = N; --i >= 0; swap(x, y)) {
		if(i+2 < N)
			for(int j : cand[C[i+2]]) x[j] = 0;

		for(int j : cand[C[i]]) 
			if((x[j] = 1 + y[(j + 1) % M]) >= M) dp[i] = 1;
	}

	int ans = 0; dp[N] = dp[0];

	for(int i = 0, j; i < N; i = j, ans++) {
		j = min(N, i + M);
		while(j > i && !dp[j]) --j;
		if(i == j) return -1;
	}
	return ans;
}

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