#684317 (C++14) No.1630 Sorting Integers (Greater than K)

• 問題: No.1630 Sorting Integers (Greater than K)
• ユーザー: ecottea
• 提出日時: 2021-07-30 21:26:40
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 5,468 bytes
• コンパイル時間: 4,010 ms
• コンパイル使用メモリ: 231,516 KB
• 実行使用メモリ: 6,588 KB
• 最終ジャッジ日時: 2024-09-15 22:57:37
• 合計ジャッジ時間: 5,510 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	44 ms; 6,460 KB
testcase_17	AC	45 ms; 6,460 KB
testcase_18	AC	43 ms; 6,332 KB
testcase_19	AC	44 ms; 6,464 KB
testcase_20	AC	39 ms; 5,376 KB
testcase_21	AC	39 ms; 5,376 KB
testcase_22	AC	38 ms; 5,376 KB
testcase_23	WA	-
testcase_24	AC	38 ms; 6,228 KB
testcase_25	AC	18 ms; 6,588 KB
testcase_26	AC	44 ms; 6,460 KB

ソースコード

#ifndef HIDDEN_IN_VISUAL_STUDIO // 無意味．折りたたむのが目的．

// 警告の抑制
#define _CRT_SECURE_NO_WARNINGS

// 使えるライブラリの読み込み
#include <bits/stdc++.h>
#include <functional> // function
using namespace std;

// 型名の短縮
using ll = long long;           // -2^63 ～ 2^63 = 9   * 10^18
using ull = unsigned long long; //     0 ～ 2^64 = 1.8 * 10^19
using uint = unsigned int;      //     0 ～ 2^32 = 4   * 10^9
using pii = pair<int, int>;
using pll = pair<ll, ll>;

// 定数の定義
const double PI = 3.141592653589793238462643383279; // 円周率
const double DEG = PI / 180.0; // θ [deg] = θ * DEG [rad]
const vector<int> dx4 = { 1, 0, -1, 0 }; // 4 近傍
const vector<int> dy4 = { 0, 1, 0, -1 };
const vector<int> dx8 = { 1, 1, 0, -1, -1, -1, 0, 1 }; // 8 近傍
const vector<int> dy8 = { 0, 1, 1, 1, 0, -1, -1, -1 };
const ll INFL = (ll)1e18;
const int INF = (int)1e9;
const double EPS = 1e-10; // 許容誤差に応じて調整

// 汎用マクロの定義
#define all(a) (a).begin(), (a).end()
#define sz(x) ((int)(x).size())
#define rep(i, n) for(int i = 0, i##_len = ((int)(n)); i < i##_len; ++i) // 0 から n-1 まで
#define repi(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i <= i##_end; ++i) // s から t まで昇順
#define repir(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i >= i##_end; --i) // s から t まで降順
#define repe(i, a) for(const auto& i : (a)) // a の全要素
#define repb(i, d) for(int i = 0, i##_len = (1 << (int)(d)); i < i##_len; ++i) // d ビット全探索
#define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て
#define repit(it, a) for(auto it = (a).begin(); it != (a).end(); it++) // イテレータを回す
#define Yes(b) if(b){cout << "Yes" << endl;}else{cout << "No" << endl;}
#define Tak(b) if(b){cout << "Takahashi" << endl;}else{cout << "Aoki" << endl;}
#define norm hypot // ノルム（2D, 3D）

// 汎用関数の定義
inline ll pow(ll n, int k) { ll v = 1; rep(i, k) v *= n; return v; } // 工夫が必要なほど k が大きかったらどうせオーバーフローするからこれでいい
inline ll pow(int n, int k) { ll v = 1; rep(i, k) v *= (ll)n; return v; }
template <typename T> inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新（更新されたら true を返す）
template <typename T> inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新（更新されたら true を返す）

// 入出力用の >>, << のオーバーロード
template <typename T, typename U> ostream& operator<< (ostream& os, pair<T, U> p) { os << "(" << p.first << "," << p.second << ")"; return os; } // pair の出力用
template <typename T, typename U> istream& operator>> (istream& is, pair<T, U>& p) { T tmp1; U tmp2; is >> tmp1 >> tmp2; p = { tmp1, tmp2 }; return is; } // pair の入力用

// 手元環境（Visual Studio）
#ifdef _MSC_VER
#define popcount (int)__popcnt // 全ビットにおける 1 の個数
#define popcountll (int)__popcnt64
inline int ctz(uint n) { unsigned long i; _BitScanForward(&i, n); return i; } // 下位ビットに並ぶ 0 の個数
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; } // 最大公約数
#define dump(a) cerr << "[DEBUG] " << endl << a << endl; // デバッグ出力用
#define dump_array(a) cerr << "[DEBUG]" << endl; for (auto x : a) {cout << x << " ";} cout << endl;
#define dump_array2(a) cerr << "[DEBUG]" << endl; for (auto c : a) {for (auto x : c) {cout << x << " ";} cout << endl;}
// 提出用（GCC）
#else
#define popcount (int)__builtin_popcount
#define popcountll (int)__builtin_popcountll
#define ctz __builtin_ctz
#define gcd __gcd
#define dump(a) 
#define dump_array(a) 
#define dump_array2(a) 
#endif

#endif // 無意味．折りたたむのが目的．


// AtCoder 専用
#include <atcoder/all>
using namespace atcoder;

// mint で使いたい法によってここを切り替える
using mint = modint1000000007;
//using mint = modint998244353;
//using mint = modint; // modint::set_mod(10000); // mint の法の指定

ostream& operator<< (ostream& os, mint x) { os << x.val(); return os; } // mint の出力用
istream& operator>> (istream& is, mint& x) { ll tmp; is >> tmp; x = tmp; return is; } // mint の入力用


int main() {
	cout << fixed << setprecision(15); // 小数点以下の桁数の指定

	int n;
	string k;
	cin >> n >> k;

	vector<int> cnt(10);
	repi(i, 1, 9) {
		cin >> cnt[i];
	}

	vector<int> up;

	function<void()> owari = [&]() {
		repe(v, up) {
			cout << v;
		}

		repi(i, 1, 9) {
			rep(hoge, cnt[i]) {
				cout << i;
			}
		}

		cout << endl;

		return 0;
	};

	if (sz(k) < n) {
		owari();
		return 0;
	}
	
	rep(i, n) {
		int j = k[i] - '0';
		if (j == 0) {
			owari();
			return 0;
		}

		while (j <= 9 && cnt[j] == 0) {
			j++;
		}

		if (j <= 9 && j > k[i] - '0') {
			up.push_back(j);
			cnt[j]--;
			owari();
			return 0;
		}

		if (j == 10 || (i == n - 1 && j == k[i] - '0')) {
		LOOP_START:

			i--;

			if (i == -1) {
				cout << -1 << endl;
				return 0;
			}

			cnt[up[i]]++;
			j = up[i] + 1;
			up.pop_back();

			while (j <= 9 && cnt[j] == 0) {
				j++;
			}

			if (j == 10) {
				goto LOOP_START;
			}

			up.push_back(j);
			cnt[j]--;
			owari();
			return 0;
		}
		else {
			up.push_back(j);
			cnt[j]--;
		}
	}

	owari();
}