#include<cstdio>
#include<iostream>
#include<sstream>
#include<cmath>
#include<algorithm>
#include<map>
#include<set>
#include<list>
#include<vector>
#include<stack>
#include<queue>
#include<string>
#include<iomanip>
#include<fstream>
#include<ctime>
#include<climits>
using namespace std;
typedef vector<int> VI;
typedef pair <int,int> ii;
typedef long long LL;
#define pb push_back
const LL INFLL = 1000000000000000001LL;
const int MA = 250005;
const int MB = 530000;
vector<LL> wek[MA + 5];
int i, j, l, tab[MA], n, c, a, w[MB], M;
LL res, add, b, k, sum, maxx;
bool overf;
int getAndErase(int e) {
int v = 1;
while (v <= M) {
w[v]--;
if (w[2 * v] >= e) v *= 2; else {
e -= w[2 * v];
v = v * 2 + 1;
}
}
w[v]--;
return v - M;
}
void init (int a) {
int i;
M=1;
while (M<a) M*=2;
for (i=1;i<2*M;i++) w[i] = 0;
M--;
for (i=1;i<=a;i++) w[M + i] = 1;
for (i=M;i>=1;i--) w[i] = w[2 * i] + w[2 * i + 1];
}
LL get(int a, LL b) {
if (b < wek[a].size()) return wek[a][b];
LL otB = a * 1LL * (a - 1) / 2 - b;
if (otB < wek[a].size()) return wek[a][otB];
return INFLL;
}
int main() {
cin >> n >> k;
b = n * 1LL * (n - 1) / 2;
if (b % 2 == 1) {
printf("NIE\n");
return 0;
}
b /= 2;
wek[1].pb(1);
for (i=2;i<=n;i++) {
overf = false;
for (j=0;j<=i*1LL*(i-1)/2;j++) {
res = 0;
for (l=0;l<i && l<=j;l++) {
add = (j - l >= wek[i - 1].size()) ? 0 : wek[i - 1][j - l];
if (res + add >= INFLL) {
overf = true;
break;
}
res += add;
}
if (overf) {
wek[i].pb(INFLL);
break;
}
wek[i].pb(res);
}
}
if (b < wek[n].size() && wek[n][b] < k) {
printf("NIE\n");
return 0;
}
init(n);
a = n;
while (a > 1) {
//cout << a << " " << b << " " << k << endl;
sum = 0;
c = 0;
maxx = (a - 1) * 1LL * (a - 2) / 2;
if (b > maxx) {
c += (b - maxx);
b = maxx;
}
while (true) {
add = get(a - 1, b);
if (sum + add >= k) {
break;
}
sum += add;
c++;
b--;
}
tab[n - a] = getAndErase(c + 1);
a--;
k-=sum;
}
tab[n - 1] = getAndErase(1);
printf("TAK\n");
for (i=0;i<n;i++) printf("%d ", tab[i]);
printf("\n");
//for (i=0;i<n - 1;i++) if (tab[i] > tab[i+1]) printf("%d\n", i);
}
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 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 | #include<cstdio> #include<iostream> #include<sstream> #include<cmath> #include<algorithm> #include<map> #include<set> #include<list> #include<vector> #include<stack> #include<queue> #include<string> #include<iomanip> #include<fstream> #include<ctime> #include<climits> using namespace std; typedef vector<int> VI; typedef pair <int,int> ii; typedef long long LL; #define pb push_back const LL INFLL = 1000000000000000001LL; const int MA = 250005; const int MB = 530000; vector<LL> wek[MA + 5]; int i, j, l, tab[MA], n, c, a, w[MB], M; LL res, add, b, k, sum, maxx; bool overf; int getAndErase(int e) { int v = 1; while (v <= M) { w[v]--; if (w[2 * v] >= e) v *= 2; else { e -= w[2 * v]; v = v * 2 + 1; } } w[v]--; return v - M; } void init (int a) { int i; M=1; while (M<a) M*=2; for (i=1;i<2*M;i++) w[i] = 0; M--; for (i=1;i<=a;i++) w[M + i] = 1; for (i=M;i>=1;i--) w[i] = w[2 * i] + w[2 * i + 1]; } LL get(int a, LL b) { if (b < wek[a].size()) return wek[a][b]; LL otB = a * 1LL * (a - 1) / 2 - b; if (otB < wek[a].size()) return wek[a][otB]; return INFLL; } int main() { cin >> n >> k; b = n * 1LL * (n - 1) / 2; if (b % 2 == 1) { printf("NIE\n"); return 0; } b /= 2; wek[1].pb(1); for (i=2;i<=n;i++) { overf = false; for (j=0;j<=i*1LL*(i-1)/2;j++) { res = 0; for (l=0;l<i && l<=j;l++) { add = (j - l >= wek[i - 1].size()) ? 0 : wek[i - 1][j - l]; if (res + add >= INFLL) { overf = true; break; } res += add; } if (overf) { wek[i].pb(INFLL); break; } wek[i].pb(res); } } if (b < wek[n].size() && wek[n][b] < k) { printf("NIE\n"); return 0; } init(n); a = n; while (a > 1) { //cout << a << " " << b << " " << k << endl; sum = 0; c = 0; maxx = (a - 1) * 1LL * (a - 2) / 2; if (b > maxx) { c += (b - maxx); b = maxx; } while (true) { add = get(a - 1, b); if (sum + add >= k) { break; } sum += add; c++; b--; } tab[n - a] = getAndErase(c + 1); a--; k-=sum; } tab[n - 1] = getAndErase(1); printf("TAK\n"); for (i=0;i<n;i++) printf("%d ", tab[i]); printf("\n"); //for (i=0;i<n - 1;i++) if (tab[i] > tab[i+1]) printf("%d\n", i); } |
English