/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 320.0 KiB
#2 Accepted 1ms 324.0 KiB
#3 Accepted 1ms 328.0 KiB
#4 Accepted 3ms 580.0 KiB
#5 Accepted 109ms 5.453 MiB
#6 Accepted 21ms 1.125 MiB
#7 Accepted 93ms 5.445 MiB
#8 Accepted 81ms 5.43 MiB

Code

/*
 *   Copyright (c) 2024 Emon Thakur
 *   All rights reserved.
 */
#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
using minheap = priority_queue<long long, vector<long long>, greater<long long>>;
typedef tree<int, null_type, greater_equal<int>, rb_tree_tag, tree_order_statistics_node_update> pbds; // find_by_order, order_of_key

#define ll long long
#define ld long double
#define MOD 1000000007
#define pie 2*(acos(0.0))
#define yes cout<<"YES\n"
#define no cout<<"NO\n"
#define pb push_back
#define endl '\n'
#define lcm(a,b) (a*b)/(__gcd<ll>(a,b))
#define print(v) for(auto e:v) cout<<e<<" "; cout<<endl;
#define printp(v) for(auto e:v) cout<<e.first<<" "<<e.second<<endl;
#define srt(v) sort(v.begin(),v.end())
#define rsrt(v) sort(v.rbegin(),v.rend())
#define life_is_a_race ios::sync_with_stdio(false); cin.tie(nullptr);

void solve(int tc)
{
    //cout<<"Case "<<tc<<": ";
    ll n, m;
    string s1, s2;
    cin >> n >> m >> s1;
    s2 = s1;
    sort(s2.begin(), s2.end());
    if(n <= m + 1) {
        cout << s2 << endl;
        return;
    }
    map<char, set<int, greater<int>>> mp;
    for(int i = 0; i < n; i++) {
        mp[s1[i]].insert(i);
    }
    char last = '0';
    priority_queue<char> pq;
    for(int i = 0; i < n && m > 0; i++) {
        if(last != s2[i]) {
            for(auto &j: mp[last]) {
                if(pq.size() == 0) break;
                s1[j] = pq.top();
                mp[s1[j]].insert(j);
                pq.pop();
            }
            last = s2[i];
        }
        if(s1[i] == s2[i]) {
            mp[s1[i]].erase(i);
            continue;
        }
        pq.push(s1[i]);
        mp[s1[i]].erase(i);
        s1[i] = s2[i];
        --m;
    }
    if(pq.size()) {
        for(auto &j: mp[last]) {
            if(pq.size() == 0) break;
            s1[j] = pq.top();
            pq.pop();
        }
    }
    cout << s1 << endl;
}
int main()
{
    life_is_a_race
    int t=1; 
    cin>>t;
    for(int i=1;i<=t;i++) solve(i);
}

Information

Submit By
Type
Submission
Problem
P1058 Lexicographically Smallest String
Contest
Sylhet ICPC 2024 Collaborative Challenge: Episode 2
Language
C++17 (G++ 13.2.0)
Submit At
2024-10-30 11:40:44
Judged At
2024-11-11 02:34:51
Judged By
Score
100
Total Time
109ms
Peak Memory
5.453 MiB