結果

問題 No.1630 Sorting Integers (Greater than K)
ユーザー ぷらぷら
提出日時 2021-07-30 21:21:13
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 30 ms / 2,000 ms
コード長 3,695 bytes
コンパイル時間 2,157 ms
コンパイル使用メモリ 196,336 KB
最終ジャッジ日時 2025-01-23 11:03:46
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int main() {
    int N;
    string K;
    cin >> N >> K;
    vector<int>tmp(9);
    for(int i = 0; i < 9; i++) {
        cin >> tmp[i];
    }
    if(N > K.size()) {
        string S = "";
        for(int i = 0; i < 9; i++) {
            S += string(tmp[i],(char)('0'+i+1));
        }
        cout << S << endl;
    }
    else if(N < K.size()) {
        cout << -1 << endl;
    }
    else {
        string S = "";
        bool flag = false;
        for(int i = 0; i < N; i++) {
            for(int j = 0; j < 9; j++) {
                char c = (char)('0'+j+1);
                if(flag) {
                    if(tmp[j]) {
                        tmp[j]--;
                        S += c;
                        break;
                    }
                }
                else {
                    if(K[i] == c && tmp[j]) {
                        tmp[j]--;
                        S += c;
                        break;
                    }
                    if(K[i] < c && tmp[j]) {
                        tmp[j]--;
                        S += c;
                        flag = true;
                        break;
                    }
                }
            }
            if(i+1 != S.size()) {
                bool ok = false;
                for(int j = i-1; j >= 0; j--) {
                    for(int k = 0; k < 9; k++) {
                        char c = (char)('0'+k+1);
                        if(tmp[k] && K[j] < c) {
                            tmp[S[j]-'0'-1]++;
                            S[j] = c;
                            tmp[k]--;
                            ok = true;
                            break;
                        }
                    }
                    if(ok) {
                        break;
                    }
                    tmp[S[j]-'0'-1]++;
                    S.pop_back();
                }
                if(!ok) {
                    cout << -1 << endl;
                    return 0;
                }
                flag = true;
                for(i = S.size(); i < N; i++) {
                    for(int j = 0; j < 9; j++) {
                        char c = (char)('0'+j+1);
                        if(flag) {
                            if(tmp[j]) {
                                tmp[j]--;
                                S += c;
                                break;
                            }
                        }
                    }
                }
                break;
            }
        }
        if(!flag) {
            bool ok = false;
            for(int j = N-1; j >= 0; j--) {
                for(int k = 0; k < 9; k++) {
                    char c = (char)('0'+k+1);
                    if(tmp[k] && K[j] < c) {
                        tmp[S[j]-'0'-1]++;
                        S[j] = c;
                        tmp[k]--;
                        ok = true;
                        break;
                    }
                }
                if(ok) {
                    break;
                }
                tmp[S[j]-'0'-1]++;
                S.pop_back();
            }
            if(!ok) {
                cout << -1 << endl;
                return 0;
            }
            flag = true;
            for(int i = S.size(); i < N; i++) {
                for(int j = 0; j < 9; j++) {
                    char c = (char)('0'+j+1);
                    if(flag) {
                        if(tmp[j]) {
                            tmp[j]--;
                            S += c;
                            break;
                        }
                    }
                }
            }
        }
        cout << S << endl;
    }
}
0