oj.uz

Submission #375340

# Submission time Handle Problem Language Result Execution time Memory
375340 2021-03-09T09:27:46 Z astoria Painting Walls (APIO20_paint) C++14
0 / 100
 1 ms 512 KB 
#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){
	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]=i;
		}
	}
	
	int lng[N+5];
	memset(lng,-1,sizeof(lng));
	lng[0]=1;
	for(int i=1; i<N; i++){
		lng[i] = 1;
		int cur = f[C[i]];
		int lst = f[C[i-1]];
		int nxt = (lst+1)%M;
		if(nxt==cur) lng[i] += lng[i-1];
	}
	
	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

paint.cpp: In function 'int minimumInstructions(int, int, int, std::vector<int>, std::vector<int>, std::vector<std::vector<int> >)':
paint.cpp:31:16: warning: comparison of integer expressions of different signedness: 'std::multiset<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   31 |   if(mns.size()>M&&i>=M) mns.erase(mns.find(dp[i-M]));
      |      ~~~~~~~~~~^~
paint.cpp:32:21: warning: comparison of integer expressions of different signedness: 'std::multiset<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   32 |   else if(mns.size()>M) mns.erase(0);
      |           ~~~~~~~~~~^~

Subtask #1
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Incorrect 1 ms 512 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #2
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Incorrect 1 ms 512 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Incorrect 1 ms 512 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #4
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Incorrect 1 ms 512 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #5
0 / 37.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Incorrect 1 ms 512 KB Output isn't correct
4 Halted 0 ms 0 KB -