Submission #296803

#	Time	Username	Problem	Language	Result	Execution time	Memory
296803		Bruteforceman	JJOOII 2 (JOI20_ho_t2)	C++11	100 / 100	18 ms	3344 KiB

ho_t2

#include "bits/stdc++.h"
using namespace std;
const int inf = 1e9;
string s;
vector <int> solve(char c, int k) {
	vector <int> v (s.size());
	int r = 0;
	int cnt = 0;
	for(int i = 0; i < s.size(); i++) {
		while(r < s.size() && cnt < k) {
			cnt += (s[r] == c);
			++r;
		}
		v[i] = (cnt == k) ? r : inf;
		cnt -= (s[i] == c);
	}	
	return v;
}
int main(int argc, char const *argv[])
{
	int n, k;
	cin >> n >> k;
	cin >> s;
	map <char, int> cnt;
	int r = 0;
	int ans = inf;
	vector <int> a[3];
	a[0] = solve('J', k);
	a[1] = solve('O', k);
	a[2] = solve('I', k); 
	for(int i = 0; i < n; i++) {
		int cur = i;
		for(int j = 0; j < 3; j++) {
			if(cur >= s.size()) {
				cur = inf;
				break;
			}
			cur = a[j][cur];
		}
		ans = min(ans, cur - i - 3 * k);
	}
 	if(ans > n) ans = -1;
 	cout << ans << endl;
	return 0;
}

Compilation message (stderr)

ho_t2.cpp: In function 'std::vector<int> solve(char, int)':
ho_t2.cpp:9:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~
ho_t2.cpp:10:11: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   10 |   while(r < s.size() && cnt < k) {
      |         ~~^~~~~~~~~~
ho_t2.cpp: In function 'int main(int, const char**)':
ho_t2.cpp:34:11: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |    if(cur >= s.size()) {
      |       ~~~~^~~~~~~~~~~
ho_t2.cpp:25:6: warning: unused variable 'r' [-Wunused-variable]
   25 |  int r = 0;
      |      ^

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