Submission #64794949

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#include <atcoder/all>
using namespace atcoder;


template<typename T>
vector<pair<T, int>> RLE(vector<T> a){
    int n = a.size();
    T ls = a[0];
    int cnt = 1;
    vector<pair<T, int>> ret;
    for(int i = 1; i < n; i++){
        if(ls != a[i]){
            ret.push_back({ls, cnt});
            ls = a[i];
            cnt = 0;
        }
        cnt++;
    }
    ret.push_back({ls, cnt});
    return ret;
}

int main(){
    int N, K; cin >> N >> K;
    vector<char> S(N);
    for(int i = 0; i < N; i++)cin >> S[i];
    int mx = 0, maru = 0;
    for(int i = 0; i < N; i++){
        if(S[i] != '?'){
            if(S[i] == 'o'){
                mx++;
                maru++;
            }
        }
        else if((i > 0 && S[i - 1] == 'o') || (i < N - 1 && S[i + 1] == 'o')){
            S[i] = '.';
        }
    }
    vector<pair<char, int>> R = RLE(S);
    for(auto [c, a] : R){
        if(c == '?')mx += (a + 1)/2;
    }
    string T = "";
    for(auto [c, a] : R){
        if(c != '?')T += string(a, c);
        else if(maru == K)T += string(a, '.');
        else if(a % 2 == 0)T += string(a, '?');
        else if(mx == K){
            for(int i = 0; i < a; i++){
                if(i % 2 == 0)T += 'o';
                else T += '.';
            }
        }
        else if(mx > K)T += string(a, '?');
    }
    cout << T << endl;
}

Submission Info

Judge Result