/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 2ms 540.0 KiB
#2 Accepted 29ms 772.0 KiB
#3 Accepted 33ms 572.0 KiB
#4 Accepted 13ms 564.0 KiB
#5 Accepted 15ms 772.0 KiB
#6 Accepted 23ms 580.0 KiB
#7 Accepted 25ms 568.0 KiB
#8 Accepted 34ms 788.0 KiB
#9 Accepted 33ms 584.0 KiB

Code

#ifndef LOCAL
#include <bits/stdc++.h>
#define debug(...)
#endif

using namespace std;
#define int long long
#define cinv(v) for (auto &it:v) cin>>it;
#define coutv(v) for (auto &it:v) cout<< it<<' '; cout<<'\n';

const int N = 2e3 + 5;
vector<bool> is_prime(N, true);
vector<int> prime{0};

void shelby() {
    int n;
    string s;
    cin >> n >> s;
    vector<int> cnt(3);
    for (auto &it: s) cnt[it - 'a']++;
    sort(cnt.begin(), cnt.end());
    debug(n, s, cnt);
    int ans = LLONG_MAX;
    auto calc = [&](int i, int j, int k)-> int {
        vector<int> v{i, j, k};
        sort(v.begin(), v.end());
        int ret = 0;
        for (int x = 0; x < 3; ++x) ret += abs(cnt[x] - v[x]);
        debug(i, j, k, ret);
        return ret / 2;
    };
    for (int i = 0; i < prime.size(); ++i) {
        for (int j = 0; j < prime.size(); ++j) {
            int k = n - prime[i] - prime[j];
            if (k >= 0 && is_prime[k]) ans = min(ans, calc(prime[i], prime[j], k));
        }
    }
    cout << (ans == LLONG_MAX ? -1 : ans) << '\n';
}

signed main() {
    cin.tie(0)->sync_with_stdio(0);
    is_prime[0] = is_prime[1] = false;
    for (int i = 2; i < N; ++i) {
        if (is_prime[i]) {
            prime.push_back(i);
            for (int j = i * i; j < N; j += i) is_prime[j] = false;
        }
    }
    int t = 1;
    cin >> t;
    for (int _ = 1; _ <= t; ++_) {
        // cout << "Case " << _ << ": ";
        shelby();
    }
}

Information

Submit By
Type
Submission
Problem
P1158 Yet another Beautiful String
Language
C++17 (G++ 13.2.0)
Submit At
2025-01-10 18:42:51
Judged At
2025-01-10 18:42:51
Judged By
Score
100
Total Time
34ms
Peak Memory
788.0 KiB