Submission #397904

#	Time	Username	Problem	Language	Result	Execution time	Memory
397904		ZikXewen	벽 칠하기 (APIO20_paint)	C++17	63 / 100	1566 ms	14244 KiB

paint

#include "paint.h"
#include <bits/stdc++.h>
using namespace std;
int minimumInstructions( int N, int M, int K, vector<int> C, vector<int> A, vector<vector<int>> B) {
	auto md = [=](int u){return ((u % M) + M) % M;};
	vector<vector<int>> f(K);
	vector<int> mem(M), an;
	for(int i = 0; i < M; ++i) for(int j: B[i]) f[j].emplace_back(i);
	for(int i = 0, ct = 0, tem; i < N; ++i) {
		if(i >= M) for(int j: f[C[i - M]]) if(--mem[md(j - i)] == M - 1) --ct;
		for(int j: f[C[i]]) if(++mem[md(j - i)] == M) ++ct;
		if(ct) an.emplace_back(i);
	}
	if(an.empty() or upper_bound(an.begin(), an.end(), M - 1) == an.begin()) return -1;
	for(int i = -1, ct = 1, tem;; i = tem, ++ct){
		tem = *(upper_bound(an.begin(), an.end(), i + M) - 1);
		if(tem == N - 1) return ct;
		if(tem == i) return -1;
	}
}

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:9:25: warning: unused variable 'tem' [-Wunused-variable]
    9 |  for(int i = 0, ct = 0, tem; i < N; ++i) {
      |                         ^~~

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