結果

問題 No.1630 Sorting Integers (Greater than K)
ユーザー eve__fuyuki
提出日時 2024-05-08 14:14:13
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 92 ms / 2,000 ms
コード長 1,660 bytes
コンパイル時間 2,164 ms
コンパイル使用メモリ 199,364 KB
最終ジャッジ日時 2025-02-21 11:44:44
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
void fast_io() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
}

int main() {
    fast_io();
    int n;
    string k;
    cin >> n >> k;
    vector<int> c(10);
    for (int i = 1; i <= 9; i++) {
        cin >> c[i];
    }
    if (n > k.size()) {
        for (int i = 1; i <= 9; i++) {
            for (int j = 0; j < c[i]; j++) {
                cout << i;
            }
        }
        cout << endl;
        return 0;
    }
    if (n < k.size()) {
        cout << -1 << endl;
        return 0;
    }
    vector<vector<int>> cum(n + 1, vector<int>(10));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < 10; j++) {
            cum[i + 1][j] = cum[i][j] + (k[i] - '0' == j);
        }
    }
    for (int i = n - 1; i >= 0; i--) {
        bool is_ok = true;
        vector<int> res(10);
        for (int j = 0; j < 10; j++) {
            res[j] = c[j] - cum[i][j];
            if (res[j] < 0) {
                is_ok = false;
                break;
            }
        }
        if (!is_ok) {
            continue;
        }
        is_ok = false;
        int l = -1;
        for (int j = (k[i] - '0') + 1; j < 10; j++) {
            if (res[j] > 0) {
                is_ok = true;
                l = j;
                break;
            }
        }
        if (!is_ok) {
            continue;
        }
        res[l]--;
        cout << k.substr(0, i);
        cout << l;
        for (int j = 1; j < 10; j++) {
            for (int m = 0; m < res[j]; m++) {
                cout << j;
            }
        }
        cout << endl;

        return 0;
    }
    cout << -1 << endl;
}
0