Submission #1111898

#	Time	Username	Problem	Language	Result	Execution time	Memory
1111898		whyalwaysmezzz	JJOOII 2 (JOI20_ho_t2)	C++14	100 / 100	12 ms	3408 KiB

ho_t2

#include <bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define FOR(i, a, b) for (int i = (a), _b = (b); i <= _b; i++)
#define FORD(i, b, a) for (int i = (b), _a = (a); i >= _a; i--)
#define BIT(mask,i) ((mask >> i) & 1ll )
#define endl '\n'  
#define all(x) x.begin(),x.end()
#define ii pair <int,int> 
using namespace std;
#define read doc() 
int doc () {bool minus = false;int result = 0;char ch;ch = getchar();while (true) {if (ch == '-') break;if (ch >= '0' && ch <= '9') break;ch = getchar();}if (ch == '-') minus = true; else result = ch-'0';while (true) {ch = getchar();if (ch < '0' || ch > '9') break;result = result*10 + (ch - '0');}if (minus)return -result;else return result;}
template<class X, class Y>bool minimize(X &x, const Y &y) {if (x > y) {x = y;return true;} else return false;}
template<class X, class Y>bool maximize(X &x, const Y &y) {if (x < y) {x = y;return true;} else return false;}
const long long oo = 1e18;
const int N = 1e6+5;
const int MOD = 1e9+7;
char a[N];
vector <int> J,O,I;
void Whyalwaysmezzz()
{
	int n,k;
	cin >> n >> k;
	FOR(i,1,n) cin >> a[i];
	FOR(i,1,n)
	{
		if (a[i] == 'J') J.push_back(i);
		if (a[i] == 'O') O.push_back(i);
		if (a[i] == 'I') I.push_back(i);
	}
	int ans = oo;
	FOR(i,0,J.size()-1)
	{
		if (i + k - 1 > (int)J.size()-1) break;
		int pos1 = J[i+k-1];
		auto it2 = lower_bound(all(O),pos1);
		if (it2 == O.end()) break;
		int pos2 = it2 - O.begin();
		if (pos2 + k - 1 > O.size()-1) break;
		// pos2=O[pos2];
		auto it3 = lower_bound(all(I),O[pos2+k-1]);
		if (it3 == I.end()) break;
		int pos3 = it3 - I.begin();
		if (pos3 + k -1 > I.size()-1) break;
		pos3 = pos3+k-1;
		// cout <<  J[i] << " " << O[pos2] << " " << I[pos3] << endl;
		ans = min(ans , I[pos3] - J[i] + 1 - k * 3); 
	}
	cout << (ans != oo ? ans : -1);
    return;
}
#define TASK "task"
signed main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);//freopen(TASK".inp","r",stdin);freopen(TASK".out","w",stdout);
    int Sty1e = 1;
    // cin >> Sty1e;
    while (Sty1e--) Whyalwaysmezzz();
    return 0;       
}

Compilation message (stderr)

ho_t2.cpp: In function 'void Whyalwaysmezzz()':
ho_t2.cpp:40:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |   if (pos2 + k - 1 > O.size()-1) break;
      |       ~~~~~~~~~~~~~^~~~~~~~~~~~
ho_t2.cpp:45:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |   if (pos3 + k -1 > I.size()-1) break;
      |       ~~~~~~~~~~~~^~~~~~~~~~~~

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