Submission #880530

#	Time	Username	Problem	Language	Result	Execution time	Memory
880530		cpptowin	JJOOII 2 (JOI20_ho_t2)	C++17	13 / 100	7 ms	10800 KiB

ho_t2

#include <bits/stdc++.h>
#define fo(i, d, c) for (int i = d; i <= c; i++)
#define fod(i, c, d) for (int i = c; i >= d; i--)
#define maxn 100010
#define N 1010
#define fi first
#define se second
#define pb emplace_back
#define en cout << "\n";
#define int long long
#define inf 1000000000
#define pii pair<int, int>
#define vii vector<pii>
#define lb(x) x & -x
#define bit(i, j) ((i >> j) & 1)
#define offbit(i, j) (i ^ (1 << j))
#define onbit(i, j) (i | (1 << j))
#define vi vector<int>
template <typename T1, typename T2>
bool minimize(T1 &a, T2 b)
{
    if (a > b)
    {
        a = b;
        return true;
    }
    return false;
}
template <typename T1, typename T2>
bool maximize(T1 &a, T2 b)
{
    if (a < b)
    {
        a = b;
        return true;
    }
    return false;
}
using namespace std;
string s;
int cnt[maxn][3];
int n;
int ans = inf;
int k;
int f[maxn][3];
main()
{
#define name "TASK"
    if (fopen(name ".inp", "r"))
    {
        freopen(name ".inp", "r", stdin);
        freopen(name ".out", "w", stdout);
    }
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n >> k >> s;
    s = ' ' + s;
    memset(f, -1, sizeof f);
    fo(i, 1, n)
    {
        fo(j, 0, 2) cnt[i][j] = cnt[i - 1][j];
        if (s[i] == 'J')
            cnt[i][0]++;
        if (s[i] == 'O')
            cnt[i][1]++;
        if (s[i] == 'I')
            cnt[i][2]++;
        if (f[cnt[i][0]][0] == -1)
            f[cnt[i][0]][0] = i;
        if (f[cnt[i][1]][1] == -1)
            f[cnt[i][1]][1] = i;
        if (f[cnt[i][2]][2] == -1)
            f[cnt[i][2]][2] = i;
    }
    // fo(i, 1, n) cout << cnt[i][0] << ' ' << cnt[i][1] << ' ' << cnt[i][2] << "\n";
    fo(i, 0, n)
    {
        // cout << i << "\n";
        int val = cnt[i][0];
        int pos = f[val + k][0];
        if (pos == -1)
            break;
        int now = 0;
        now += cnt[pos][1] - cnt[i][1] + cnt[pos][2] - cnt[i][2];
        val = cnt[pos][1];
        int pre = pos;
        pos = f[val + k][1];
        if (pos == -1)
            break;
        now += cnt[pos][2] - cnt[pre][2] + cnt[pos][0] - cnt[pre][0];
        // cout << now << ' ' << pos << "\n";
        pre = pos;
        val = cnt[pos][2];
        pos = f[val + k][2];
        if (pos == -1)
            break;
        now += cnt[pos][1] - cnt[pre][1] + cnt[pos][0] - cnt[pre][0];
        // cout << now << ' ' << pos << "\n";
        ans = min(ans, now);
        // cout << ans << "\n";
        // cout << now << "\n";
    }
    cout << (ans == inf ? -1 : ans);
}

Compilation message (stderr)

ho_t2.cpp:46:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   46 | main()
      | ^~~~
ho_t2.cpp: In function 'int main()':
ho_t2.cpp:51:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   51 |         freopen(name ".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
ho_t2.cpp:52:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   52 |         freopen(name ".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~

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