Submission #576655

#	Time	Username	Problem	Language	Result	Execution time	Memory
576655		pnm1384	Painting Walls (APIO20_paint)	C++14	100 / 100	764 ms	18656 KiB

paint

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

#define F first
#define S second

const int N = 3e5 + 20;
vector<int> vt[N];
int col[N], last[N], dp[N];
bool have[N];
vector<pair<int, int>> vt1, vt2;

int minimumInstructions(int N, int M, int K, std::vector<int> C, std::vector<int> A, std::vector<std::vector<int>> B) 
{
	int n = N, m = M, k = K;
	for (int i = 0; i < n; i++)
	{
		col[i] = C[i];
	}
	for (int i = 0; i < m; i++)
	{
		for (int j = 0; j < A[i]; j++)
		{
			int c = B[i][j];
			vt[c].push_back(i);
		}
	}
	for (int i = 0; i < m; i++)
	{
		last[i] = -2;
		dp[i] = -2;
	}
	for (int i = 0; i < n; i++)
	{
		vt1.clear(); vt2.clear();
		for (int j : vt[col[i]])
		{
			int j2 = j - 1;
			if (j2 < 0) j2 = m - 1;
			if (last[j2] == i - 1)
			{
				vt1.push_back({j, min(dp[j2] + 1, m)});
			}
			else
			{
				vt2.push_back({j, 1});
			}
		}
		for (pair<int, int> pppp : vt1)
		{
			int j = pppp.F;
			dp[j] = pppp.S;
			assert(dp[j] <= i + 1);
			last[j] = i;
		}
		for (pair<int, int> pppp : vt2)
		{
			int j = pppp.F;
			dp[j] = 1;
			last[j] = i;
		}
		for (int j : vt[col[i]])
		{
			if (dp[j] == m) 
			{
				have[i - m + 1] = true;
				//assert(i - m + 1 >= 0);
			}
		}
	}
	int i = 0, j = 0;
	int ans = 0;
	while (i < n)
	{
		int ff = -1;
		for (int kk = i; kk >= j; kk--)
		{
			if (have[kk])
			{
				ff = kk;
				break;
			}
		}
		if (ff == -1) return -1;
		ans++;
		j = i + 1;
		i = m + ff;
	}
	return ans;
}

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:16:20: warning: unused variable 'k' [-Wunused-variable]
   16 |  int n = N, m = M, k = K;
      |                    ^

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