結果

問題 No.1630 Sorting Integers (Greater than K)
ユーザー monnu
提出日時 2021-07-30 21:11:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 91 ms / 2,000 ms
コード長 1,437 bytes
コンパイル時間 4,873 ms
コンパイル使用メモリ 230,204 KB
実行使用メモリ 38,872 KB
最終ジャッジ日時 2024-09-15 22:16:57
合計ジャッジ時間 5,660 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll=long long;
using Graph=vector<vector<pair<int,int>>>;
#define MAX 100000
#define MOD 1000000007
#define INF 1000000000

int main(){
  int N;
  string K;
  cin>>N>>K;
  vector<int> c(9);
  for(int i=0;i<9;i++){
    cin>>c[i];
  }
  int n=K.size();
  if(n>N){
    cout<<-1<<endl;
  }else if(n<N){
    for(int i=0;i<9;i++){
      for(int j=0;j<c[i];j++){
        cout<<i+1;
      }
    }
    cout<<endl;
  }else{
    vector<vector<int>> sum(N+1,vector<int>(9,0));
    for(int i=0;i<N;i++){
      for(int j=0;j<9;j++){
        sum[i+1][j]=sum[i][j];
      }
      if(K[i]!='0'){
        sum[i+1][K[i]-'1']++;
      }
    }
    int i=0;
    while(i<n-1&&K[i]!='0'){
      i++;
    }
    for(;i>=0;i--){
      bool flag=true;
      for(int j=0;j<9;j++){
        if(c[j]<sum[i][j]){
          flag=false;
          break;
        }
      }
      if(!flag){
        continue;
      }
      int k=-1;
      for(int j=K[i]-'0';j<9;j++){
        if(c[j]-sum[i][j]>0){
          k=j;
          flag=false;
          break;
        }
      }
      if(flag){
        continue;
      }
      for(int j=0;j<i;j++){
        cout<<K[j];
      }
      cout<<k+1;
      c[k]--;
      for(int j=0;j<9;j++){
        for(int l=0;l<c[j]-sum[i][j];l++){
          cout<<j+1;
        }
      }
      cout<<endl;
      return 0;
    }
    cout<<-1<<endl;
  }
}
0