제출 #685093

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
685093		vovamr	Devil's Share (RMI19_devil)	C++17	100 / 100	284 ms	6924 KiB

devil

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define fi first
#define se second
#define ll long long
#define ld long double
#define sz(x) ((int)(x).size())
#define all(x) 	(x).begin(), (x).end()
#define pb push_back
#define mpp make_pair
#define ve vector
using namespace std;
using namespace __gnu_pbds;
template<class T> using oset = tree<T,null_type,less<T>,rb_tree_tag,tree_order_statistics_node_update>;
const ll inf = 1e18; const int iinf = 1e9;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
template <typename T> inline bool chmin(T& a, T b) { return (a > b ? a = b, 1 : 0); }
template <typename T> inline bool chmax(T& a, T b) { return (a < b ? a = b, 1 : 0); }

inline void solve() {
	int k;
	cin >> k;
	ve<int> c(10);
	for (int i = 1; i < 10; ++i) cin >> c[i];

	ve<int> last; int kk = k - 1;
	for (int i = 9; ~i; --i) {
		while (kk && c[i]) {
			last.pb(i);
			--kk, --c[i];
		}
	}
	reverse(all(last));

	int mx = -1;
	for (int i = 9; ~i; --i) {
		if (c[i]) {
			mx = i;
			break;
		}
	}
	if (!mx) {
		for (auto &i : last) cout << i;
		cout << '\n';
		return;
	}

	ve<ve<int>> answer(c[mx]);
	for (auto &vec : answer) vec = {mx};
	
	c[mx] = 0;
	
	queue<ve<int>> q;
	for (int i = 0; i < 10; ++i) {
		while (c[i]--) {
			q.push(ve<int>{i});
		}
	}

	int ptr = sz(answer) - 1;
	while (sz(q)) {
		ve<int> cc = q.front();
		q.pop();
		for (const auto &x : cc) answer[ptr].pb(x);

		if (ptr && (!sz(q) || q.front() != cc)) {
			while (sz(answer) > ptr) {
				q.push(answer.back());
				answer.pop_back();
			}
		}
		ptr = (!ptr ? sz(answer) - 1 : ptr - 1);
	}
	for (auto &vec : answer) {
		for (auto &x : vec) cout << x;
	}
	for (auto &x : last) cout << x;
	cout << '\n';
}

signed main() {
	ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int q = 1; cin >> q;
	while (q--) solve();
	cerr << fixed << setprecision(3) << "Time execution: " << (double)clock() / CLOCKS_PER_SEC << endl;
}

• Subtask 1: 13 / 13
• Subtask 2: 14 / 14
• Subtask 3: 29 / 29
• Subtask 4: 44 / 44