Submission #647329

#	Time	Username	Problem	Language	Result	Execution time	Memory
647329		Astrayt	JJOOII 2 (JOI20_ho_t2)	C++17	100 / 100	12 ms	3288 KiB

ho_t2

//君の手を握ってしまったら
//孤独を知らないこの街には
//もう二度と帰ってくることはできないのでしょう
//君が手を差し伸べた 光で影が生まれる
//歌って聞かせて この話の続き
//連れて行って見たことない星まで
//さユリ - 花の塔
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define int ll
#define starburst ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define pii pair<int,int>
#define pb push_back
#define ff first
#define ss second
#define N 100005

void solve(){
    int n, k; cin >> n >> k;
    string s; cin >> s;
    vector<int> jp, op, ip;
    for(int i = 0; i < n; ++i){
        if(s[i] == 'J') jp.pb(i);
        if(s[i] == 'O') op.pb(i);
        if(s[i] == 'I') ip.pb(i);
    }
    if(jp.size() < k || op.size() < k || ip.size() < k) {
        cout << "-1\n";
        return;
    }
    int ans = 1e18;
    for(int i = 0, j = k - 1; j < op.size(); ++i, ++j){
        int lo = op[i], ro = op[j];
        int rj = (upper_bound(jp.begin(), jp.end(), lo) - jp.begin() - 1), lj = rj - k + 1;
        if(lj < 0) continue;
        int li = (upper_bound(ip.begin(), ip.end(), ro) - ip.begin()), ri = li + k - 1;
        if(ri >= ip.size()) continue;
        ans = min(ip[ri] - jp[lj] + 1 - 3 * k, ans);
    }
    cout << (ans == 1e18 ? -1 : ans) << '\n';
}

signed main(){
    starburst
    int t = 1; //cin >> t;
    while(t--) solve();
}

Compilation message (stderr)

ho_t2.cpp: In function 'void solve()':
ho_t2.cpp:28:18: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   28 |     if(jp.size() < k || op.size() < k || ip.size() < k) {
      |        ~~~~~~~~~~^~~
ho_t2.cpp:28:35: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   28 |     if(jp.size() < k || op.size() < k || ip.size() < k) {
      |                         ~~~~~~~~~~^~~
ho_t2.cpp:28:52: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   28 |     if(jp.size() < k || op.size() < k || ip.size() < k) {
      |                                          ~~~~~~~~~~^~~
ho_t2.cpp:33:33: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |     for(int i = 0, j = k - 1; j < op.size(); ++i, ++j){
      |                               ~~^~~~~~~~~~~
ho_t2.cpp:38:15: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   38 |         if(ri >= ip.size()) continue;
      |            ~~~^~~~~~~~~~~~

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