Submission #628510

#	Time	Username	Problem	Language	Result	Execution time	Memory
628510		Karliver	JJOOII 2 (JOI20_ho_t2)	C++17	100 / 100	17 ms	2796 KiB

ho_t2

	
#include <bits/stdc++.h>

#define FIXED_FLOAT(x)  std::fixed <<std::setprecision(20) << (x)
#define all(v) (v).begin(), (v).end()
using namespace  std;
#define forn(i,n) for (int i = 0; i < (n); ++i)
#define rforn(i, n) for(int i = (n) - 1;i >= 0;--i)
#define sz(x) (int)x.size()
#define ff first
#define se second
#define mp make_pair
using ll = long long;
int mod = (ll)1e9 + 7;
const int INF = 1e9 + 1;
const int N = 2e5 + 100;
const double eps = 1e-7;

template <class T> using V = vector<T>;  
template <class T> using VV = V<V<T>>;  
template<class T, size_t SZ> using AR = array<T, SZ>;
template<class T> using PR = pair<T, T>;
template <typename XPAX>
bool ckma(XPAX &x, XPAX y) {
    return (x < y ? x = y, 1 : 0);
}
template <typename XPAX>
bool ckmi(XPAX &x, XPAX y) {
    return (x > y ? x = y, 1 : 0);
}



void __print(int x) {cerr << x;}
void __print(long x) {cerr << x;}
void __print(long long x) {cerr << x;}
void __print(unsigned x) {cerr << x;}
void __print(unsigned long x) {cerr << x;}
void __print(unsigned long long x) {cerr << x;}
void __print(float x) {cerr << x;}
void __print(double x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '\"' << x << '\"';}
void __print(const string &x) {cerr << '\"' << x << '\"';}
void __print(bool x) {cerr << (x ? "true" : "false");}

template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ','; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? "," : ""), __print(i); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}
#define debug(x...) cerr << "[" << #x << "] = ["; _print(x)

	
void solve() {
	
	int n, k;
	cin >> n >> k;
	string s;
	cin >> s;
	
	vector<int> a(n);
	
	for(int i = 0;i < n;++i) {
		a[i] = (s[i] == 'J' ? 0 : (s[i] == 'O' ? 1 : 2));
	}
	for(int j = 0;j < 3;++j) {
		if(count(all(a), j) < k) {
			cout << -1 << '\n';
			return;
		}
	} 
	vector<int> p[3];
	for(int i = 0;i < n;++i) p[a[i]].push_back(i);
	int res = n + 1; 
	for(int i = 0;i < n;++i) {
		if(a[i] != 0)continue;
		int p1 = lower_bound(all(p[0]), i)- begin(p[0]);
		if(p1+k-1>= size(p[0])) {
			continue;
		}
		p1 = p[0][p1+k-1];
		int p2 = lower_bound(all(p[1]), p1)-begin(p[1]);
		if(p2+k-1>=size(p[1]))continue;
		p2 = p[1][p2+k-1];
		int p3 = lower_bound(all(p[2]), p2)-begin(p[2]);
		if(p3+k-1>= size(p[2]))continue;
		p3 = p[2][p3+k-1];
		res = min(res, p3-i+1-k*3);
	}
	
	cout << (res == n + 1 ? -1 : res) << '\n';
	
	


}
void test_case() {
    int t;
    cin >> t;
    forn(p, t) {

        //cout << "Case #" << p + 1 << ": ";
        solve();
    }
}
int main() {

    ios::sync_with_stdio(false);
    cin.tie(0);

    solve();

}

Compilation message (stderr)

ho_t2.cpp: In function 'void solve()':
ho_t2.cpp:82:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   82 |   if(p1+k-1>= size(p[0])) {
      |      ~~~~~~^~~~~~~~~~~~~
ho_t2.cpp:87:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   87 |   if(p2+k-1>=size(p[1]))continue;
      |      ~~~~~~^~~~~~~~~~~~
ho_t2.cpp:90:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   90 |   if(p3+k-1>= size(p[2]))continue;
      |      ~~~~~~^~~~~~~~~~~~~

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