Submission #375348

#	Time	Username	Problem	Language	Result	Execution time	Memory
375348		astoria	Painting Walls (APIO20_paint)	C++14	0 / 100	6 ms	5356 KiB

paint

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

int minimumInstructions(int N, int M, int K, std::vector<int> C, std::vector<int> A, std::vector<std::vector<int>> B){
	vector<int> f[K+5]; //for the colour i, who can do?
	for(int i=0; i<M; i++){
		for(int j=0; j<A[i]; j++){
			int col = B[i][j];
			f[col].push_back(i);
		}
	}
	
	int lng[N+5];
	memset(lng,-1,sizeof(lng));
	for(int j : f[C[0]]) lng[0]=1;
	lng[0]=max(lng[0],0);
	
	for(int i=1; i<N; i++){
		for(int j : f[C[i]]){
			lng[i] = 1;
			int lst = j-1; if(lst<0) lst+=M;
			if(f[C[i-1]].front()==lst) lng[i] += lng[i-1];
		}
		lng[i]=max(lng[i],0);
	}
	
	int dp[N+5];
	multiset<int> mns; mns.insert(0);
	for(int i=0; i<N; i++){
		if(lng[i]<M) dp[i]=1e9;
		else dp[i]=(*mns.begin())+1;
		mns.insert(dp[i]);
		if(mns.size()>M&&i>=M) mns.erase(mns.find(dp[i-M]));
		else if(mns.size()>M) mns.erase(0);
	}
	if(dp[N-1]>=1e6) return -1;
	return dp[N-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:16:10: warning: unused variable 'j' [-Wunused-variable]
   16 |  for(int j : f[C[0]]) lng[0]=1;
      |          ^
paint.cpp:34:16: warning: comparison of integer expressions of different signedness: 'std::multiset<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   34 |   if(mns.size()>M&&i>=M) mns.erase(mns.find(dp[i-M]));
      |      ~~~~~~~~~~^~
paint.cpp:35:21: warning: comparison of integer expressions of different signedness: 'std::multiset<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   35 |   else if(mns.size()>M) mns.erase(0);
      |           ~~~~~~~~~~^~

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