/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 532.0 KiB
#2 Accepted 6ms 532.0 KiB
#3 Accepted 7ms 532.0 KiB
#4 Accepted 7ms 532.0 KiB
#5 Accepted 8ms 532.0 KiB
#6 Accepted 11ms 700.0 KiB
#7 Accepted 5ms 592.0 KiB
#8 Accepted 6ms 696.0 KiB
#9 Accepted 14ms 1.277 MiB
#10 Accepted 14ms 1.227 MiB
#11 Accepted 6ms 788.0 KiB
#12 Accepted 6ms 788.0 KiB
#13 Accepted 14ms 1.77 MiB
#14 Accepted 4ms 836.0 KiB

Code

// Authored by Ibrahimfostok...
// Next level : "Master"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 998244353;
// const ll mod = 1e9 + 7;
#define gcd __gcd
#define int ll
#define ld long double
#define lcm(a, b) (a * b / gcd(a, b))
#define ceil(x, y) (((x) + (y) - 1ll) / (y))
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define pb push_back
#define MOD(x) x = ((x % mod) + mod) % mod
const long double pi = 3.14159265358979323846;
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
ll rand(ll L, ll R)
{
    return uniform_int_distribution<ll>(L, R)(rng);
}
//
void My_Solve(int TC)
{
    // cout << setprecision(10) << fixed;
    int n, k;
    cin >> n >> k;
    string s;
    cin >> s;
    int cnt = 0;
    vector<int> a;
    for (int i = 0; i < n; i++)
    {
        if (s[i] == '0')
        {
            if (cnt)
                a.push_back(cnt), cnt = 0;
        }
        else
        {
            cnt++;
        }
    }
    if (cnt)
        a.push_back(cnt);
    int ans = 0;
    sort(rall(a));
    for (int i = 0; i < min((int)a.size(), k + 1); i++)
        ans += a[i];
    cout << ans << '\n';
}
int32_t main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    // freopen("input.txt", "r", stdin);
    // freopen("output.txt", "w", stdout);
    int t = 1;
    cin >> t;
    for (int i = 1; i <= t; i++)
        My_Solve(i);
    return 0;
}

Information

Submit By
Type
Submission
Problem
P1159 Binary String
Contest
Brain Booster #8
Language
C++17 (G++ 13.2.0)
Submit At
2025-02-17 15:07:43
Judged At
2025-02-17 15:07:43
Judged By
Score
100
Total Time
14ms
Peak Memory
1.77 MiB