Submission #286151

#	Time	Username	Problem	Language	Result	Execution time	Memory
286151		reymontada61	JJOOII 2 (JOI20_ho_t2)	C++14	100 / 100	43 ms	2056 KiB

ho_t2

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

int n, k;
string s;
vector<int> tr[3];

int start(int at) {
	int jind = lower_bound(tr[0].begin(), tr[0].end(), at) - tr[0].begin();
	if (jind + k >= tr[0].size()) return INT_MAX;
	int oind = lower_bound(tr[1].begin(), tr[1].end(), tr[0][jind+k]) - tr[1].begin();
	if (oind + k >= tr[1].size()) return INT_MAX;
	int iind = lower_bound(tr[2].begin(), tr[2].end(), tr[1][oind+k]) - tr[2].begin();
	if (iind + k >= tr[2].size()) return INT_MAX;
	return tr[2][iind+k];
}

signed main() {

	cin >> n >> k;
	cin >> s;
	k--;

	for (int i=1; i<=n; i++) {
		if (s[i-1] == 'J') tr[0].push_back(i);
		if (s[i-1] == 'O') tr[1].push_back(i);
		if (s[i-1] == 'I') tr[2].push_back(i);
	}
	
	int le = INT_MAX;
	
	for (int i=1; i<=n; i++) {
		le = min(le, start(i) - i + 1);
	}
	
	if (le > n) {
		cout << -1 << endl;
	} 
	else {
		cout << le - 3 * (k+1) << endl;
	}

}

Compilation message (stderr)

ho_t2.cpp: In function 'int start(int)':
ho_t2.cpp:10:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   10 |  if (jind + k >= tr[0].size()) return INT_MAX;
      |      ~~~~~~~~~^~~~~~~~~~~~~~~
ho_t2.cpp:12:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 |  if (oind + k >= tr[1].size()) return INT_MAX;
      |      ~~~~~~~~~^~~~~~~~~~~~~~~
ho_t2.cpp:14:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   14 |  if (iind + k >= tr[2].size()) return INT_MAX;
      |      ~~~~~~~~~^~~~~~~~~~~~~~~

• Subtask 1: 1 / 1
• Subtask 2: 12 / 12
• Subtask 3: 87 / 87