Submission #563411

#	Time	Username	Problem	Language	Result	Execution time	Memory
563411		8e7	Painting Walls (APIO20_paint)	C++17	40 / 100	234 ms	30948 KiB

paint

//Challenge: Accepted
#include "paint.h"
#include <bits/stdc++.h>
using namespace std;
#ifdef zisk
void debug(){cout << endl;}
template<class T, class ... U> void debug(T a, U ... b){cout << a << " ", debug(b...);}
template<class T> void pary(T l, T r) {
	while (l != r) cout << *l << " ", l++;
	cout << endl;
}
#else
#define debug(...) 0
#define pary(...) 0
#endif
#define ll long long
#define maxn 200005
#define pii pair<int, int>
#define ff first
#define ss second
#define io ios_base::sync_with_stdio(0);cin.tie(0);
vector<int> f[maxn];
int n, m;
int zeros = 0;
int counter = 0;
int num[maxn];
void ch(int col, int ind, int d) {
	for (int i:f[col]) {
		int p = ((i - ind) % m + m)%m;
		if (num[p] == 0) zeros--;
		num[p] += d;
		if (num[p] == 0) zeros++;
	}
}
vector<int> pos;
bool poss[maxn];
int to[18][2 * maxn];

int minimumInstructions(int N, int M, int K, vector<int> C, vector<int> A, vector<vector<int>> B) {
	counter++;
	assert(counter == 1);
	m = M;
	n = N;
	for (int i = 0;i < M;i++) {
		for (int j = 0;j < A[i];j++) {
			f[B[i][j]].push_back(i);
		}
	}
	for (int i = 0;i < N;i++) {
		C.push_back(C[i]);
	}
	for (int i = 0;i < m;i++) num[i] = m;
	for (int i = 0;i < m;i++) ch(C[i], i, -1);
	for (int i = 0;i < n;i++) {
		//debug(i, zeros);
		//pary(num, num + m);

		poss[i] = zeros > 0;	
		if (poss[i]) pos.push_back(i);
		ch(C[i], i, 1);
		ch(C[i+m], i+m, -1);
	}
	int ps = pos.size();
	for (int i = 0;i < ps;i++) pos.push_back(pos[i] + n);	
	for (int i:pos) {
		int ind = upper_bound(pos.begin(), pos.end(), i+m) - pos.begin() - 1;
		to[0][i] = pos[ind];
	}
	for (int i = 1;i < 18;i++) {
		for (int j = 0;j < 2 * n;j++) to[i][j] = to[i-1][to[i-1][j]]; 
	}
	int ret = 1<<30;
	for (int i = 0;i < n;i++) {
		if (poss[i]) {
			int cur = i, val = 0;
			for (int j = 17;j >= 0;j--) {
				if (to[j][cur] < i + n - m) {
					val += 1<<j;
					cur = to[j][cur];
				}
			}
			if (m == n) ret = 1;	
			else if (to[0][cur] >= i + n - m) {
				debug(i, cur, val);
				ret = min(ret, val + 2);	
			}
		}
	}
	if (ret > n) return -1;
	return ret;
}

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:13:20: warning: statement has no effect [-Wunused-value]
   13 | #define debug(...) 0
      |                    ^
paint.cpp:84:5: note: in expansion of macro 'debug'
   84 |     debug(i, cur, val);
      |     ^~~~~

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