Submission #357704

#	Time	Username	Problem	Language	Result	Execution time	Memory
357704		Mlxa	Permutation Recovery (info1cup17_permutation)	C++14	43 / 100	4072 ms	34284 KiB

Main

#ifdef LC
#include "pch.h"
#else
#include <bits/stdc++.h>
#endif
using namespace std;
#define all(x) x.begin(), x.end()
#define x first
#define y second
#define mp make_pair
#define mt make_tuple

using num = vector<int>;

num sub(num a, num b) {
	// for (int e : a) {
	// 	cout << e;
	// }
	// cout << endl;
	// for (int e : b) {
	// 	cout << e;
	// }
	// cout << endl;
	assert(a.size() >= b.size());
	int carry = 0;
	for (int i = 0; i < (int)a.size(); ++i) {
		carry += a[i];
		if (i < (int)b.size()) {
			carry -= b[i];
		}
		while (carry < 0) {
			assert(i + 1 < (int)a.size());
			carry += 10;
			a[i + 1] -= 1;
		}
		assert(carry < 10);
		a[i] = carry;
		carry = 0;
	}
	while (a.size() >= 2 && !a.back()) {
		a.pop_back();
	}
	return a;
}

const int N = 1e5;
int n;
num dp[N];
int a[N];
vector<int> pos;

signed main() {
#ifdef LC
	assert(freopen("input.txt", "r", stdin));
#endif
	ios::sync_with_stdio(0); cin.tie(0);
	cin >> n;
	for (int i = 0; i < n; ++i) {
		string s;
		cin >> s;
		dp[i].clear();
		reverse(all(s));
		for (char c : s) {
			dp[i].push_back(c - '0');
		}
	}
	for (int i = n - 1; i >= 1; --i) {
		dp[i] = sub(dp[i], dp[i - 1]);
	}
	for (int i = 0; i < n; ++i) {
		int ind = 0;
		num least = dp[i];
		while (ind < (int)pos.size() && (least.size() >= 2 || least.front() != 1)) {
			least = sub(least, dp[pos[ind]]);
			++ind;
		}
		pos.insert(pos.begin() + ind, i);
	}
	for (int i = 0; i < n; ++i) {
		a[pos[i]] = i + 1;
	}
	for (int i = 0; i < n; ++i) {
		cout << a[i] << " ";
	}
	cout << endl;
	return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 15 / 15
• Subtask 3: 18 / 18
• Subtask 4: 0 / 17
• Subtask 5: 0 / 11
• Subtask 6: 0 / 7
• Subtask 7: 0 / 16
• Subtask 8: 0 / 6