#include <iostream>
#include <cmath>
#define LL long long
#define MAX_SIZE 250010
using namespace std;
LL get_max_inversions_count(int n) {
LL max_inv = 0;
for (LL i = 0; i < n; i++) {
max_inv += i;
}
return max_inv;
}
LL get_permutations_count(int n, LL inv) {
LL max_inv, count = 0, value;
int m;
if (inv == 0) {
if (n < 2)
return 0LL;
return 1LL;
}
max_inv = get_max_inversions_count(n);
if (inv > max_inv || max_inv == 0)
return 0LL;
if (inv == max_inv)
return 1LL;
m = n - 1;
for (LL i = max(0LL, inv-m); i <= inv; i++) {
value = get_permutations_count(m, i);
count += value;
}
return count;
}
void select_item(int pos, int n, LL k, LL inv, int *result) {
LL start = 0, max_perms, added_inv = 0;
int m = n - pos - 1;
for (int i = pos; i < n; i++ ) {
max_perms = get_permutations_count(m, inv);
if (k <= max_perms || inv == 0) {
for (int j = i; j > pos; j--) {
swap(result[j-1], result[j]);
}
if (pos < n - 1)
select_item(pos+1, n, k, inv, result);
break;
}
inv--;
k -= max_perms;
}
}
int main()
{
int n, result[MAX_SIZE];
LL k, inv, max_inv, stable_inv;
cin >> n >> k;
max_inv = get_max_inversions_count(n);
if (max_inv % 2 == 1) {
cout << "NIE" << endl;
return 0;
}
inv = max_inv / 2;
stable_inv = get_permutations_count(n, inv);
if (k > stable_inv) {
cout << "NIE" << endl;
return 0;
}
cout << "TAK" << endl;
for (int i = 0; i < n; i++)
result[i] = i + 1;
select_item(0, n, k, inv, result);
for (int i = 0; i < n; i++)
cout << result[i] << " ";
cout << endl;
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | #include <iostream> #include <cmath> #define LL long long #define MAX_SIZE 250010 using namespace std; LL get_max_inversions_count(int n) { LL max_inv = 0; for (LL i = 0; i < n; i++) { max_inv += i; } return max_inv; } LL get_permutations_count(int n, LL inv) { LL max_inv, count = 0, value; int m; if (inv == 0) { if (n < 2) return 0LL; return 1LL; } max_inv = get_max_inversions_count(n); if (inv > max_inv || max_inv == 0) return 0LL; if (inv == max_inv) return 1LL; m = n - 1; for (LL i = max(0LL, inv-m); i <= inv; i++) { value = get_permutations_count(m, i); count += value; } return count; } void select_item(int pos, int n, LL k, LL inv, int *result) { LL start = 0, max_perms, added_inv = 0; int m = n - pos - 1; for (int i = pos; i < n; i++ ) { max_perms = get_permutations_count(m, inv); if (k <= max_perms || inv == 0) { for (int j = i; j > pos; j--) { swap(result[j-1], result[j]); } if (pos < n - 1) select_item(pos+1, n, k, inv, result); break; } inv--; k -= max_perms; } } int main() { int n, result[MAX_SIZE]; LL k, inv, max_inv, stable_inv; cin >> n >> k; max_inv = get_max_inversions_count(n); if (max_inv % 2 == 1) { cout << "NIE" << endl; return 0; } inv = max_inv / 2; stable_inv = get_permutations_count(n, inv); if (k > stable_inv) { cout << "NIE" << endl; return 0; } cout << "TAK" << endl; for (int i = 0; i < n; i++) result[i] = i + 1; select_item(0, n, k, inv, result); for (int i = 0; i < n; i++) cout << result[i] << " "; cout << endl; return 0; } |
English