結果

問題 No.1630 Sorting Integers (Greater than K)
ユーザー H3PO4
提出日時 2023-10-09 10:50:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,510 bytes
コンパイル時間 768 ms
コンパイル使用メモリ 74,128 KB
最終ジャッジ日時 2025-02-17 06:24:25
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 11 TLE * 1 -- * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <array>
#include <iostream>
#include <string>
#include <vector>

std::string solve(int N, std::string K, std::array<int, 9> C) {
    if (K.size() > N) return "-1";
    if (K.size() < N) {
        // 一番小さくなるように作る
        std::string res;
        for (int i = 1; i < 10; i++) {
            res += std::string(C.at(i - 1), char(i) + '0');
        }
        return res;
    }
    for (int i = K.size() - 1; i >= 0; i--) {
        std::array<int, 9> used;
        for (int j = 0; j < 9; j++) used.at(j) = C.at(j);

        bool flag = true;
        for (int j = 0; j < i; j++) {
            int k = (int)K.at(j) - '0';
            if (k == 0) {
                flag = false;
                break;
            };
            used.at(k - 1)--;
            if (used.at(k - 1) < 0) {
                flag = false;
                break;
            }
        }
        if (!flag) continue;

        for (int n = (int)K.at(i) - '0' + 1; n < 10; n++) {
            if (used.at(n - 1) <= 0) continue;
            used.at(n - 1)--;
            std::string res = K.substr(0, i);
            res += (char)n + '0';
            for (int i = 1; i < 10; i++) {
                res += std::string(used.at(i - 1), char(i) + '0');
            }
            return res;
        }
    }
    return "-1";
}

int main() {
    int N;
    std::string K;
    std::array<int, 9> C;
    std::cin >> N >> K;
    for (int i = 0; i < 9; i++) std::cin >> C.at(i);
    std::cout << solve(N, K, C) << std::endl;
}
0