#include <bits/stdc++.h> using namespace std; #define rep(i,a,n) for (int i=a;i<n;i++) #define per(i,a,n) for (int i=n-1;i>=a;i--) #define pb push_back #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define SZ(x) ((int)(x).size()) typedef vector<int> VI; typedef long long ll; typedef pair<int,int> PII; const ll mod=1000000007; ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;} ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;} // head const int N=251000; const ll inf=1000000000001000000ll; vector<ll> dp[N]; int n; ll k; void init() { dp[1].pb(1); for (int i=2;i<=n;i++) { ll w=0; bool full=SZ(dp[i-1])==(ll)(i-1)*(i-2)/2+1; for (int j=0;j<=(ll)i*(i-1)/2;j++) { if (j>=SZ(dp[i-1])&&!full) break; if (j<SZ(dp[i-1])) w+=dp[i-1][j]; if (j>=i) w-=dp[i-1][j-i]; if (w>inf) break; dp[i].pb(w); } } } ll count(int n,ll k) { if (n==0) return k==0; k=min(k,(ll)n*(n-1)/2-k); if (k<0) return 0; if (k>=SZ(dp[n])) return inf; else return dp[n][k]; } set<int> s; void gao(int n,ll inv) { for (int i=0;i<n;i++) s.insert(i+1); for (int i=0;i<n;i++) { ll sz=(ll)(n-i-1)*(n-i-2)/2; for (int j=max(0ll,inv-sz);j<n-i;j++) { if (count(n-i-1,inv-j)<k) { k-=count(n-i-1,inv-j); } else { inv-=j; auto it=s.begin(); if (2*j<SZ(s)) { rep(k,0,j) ++it; } else { it=s.end(); rep(k,j,SZ(s)) --it; } printf("%d ",*it); s.erase(it); break; } } } puts(""); } int main() { scanf("%d%lld",&n,&k); if ((ll)n*(n-1)%4!=0) { puts("NIE"); return 0; } init(); if (count(n,(ll)n*(n-1)/4)<k) { puts("NIE"); return 0; } puts("TAK"); ll inv=(ll)n*(n-1)/4; gao(n,inv); }
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 | #include <bits/stdc++.h> using namespace std; #define rep(i,a,n) for (int i=a;i<n;i++) #define per(i,a,n) for (int i=n-1;i>=a;i--) #define pb push_back #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define SZ(x) ((int)(x).size()) typedef vector<int> VI; typedef long long ll; typedef pair<int,int> PII; const ll mod=1000000007; ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;} ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;} // head const int N=251000; const ll inf=1000000000001000000ll; vector<ll> dp[N]; int n; ll k; void init() { dp[1].pb(1); for (int i=2;i<=n;i++) { ll w=0; bool full=SZ(dp[i-1])==(ll)(i-1)*(i-2)/2+1; for (int j=0;j<=(ll)i*(i-1)/2;j++) { if (j>=SZ(dp[i-1])&&!full) break; if (j<SZ(dp[i-1])) w+=dp[i-1][j]; if (j>=i) w-=dp[i-1][j-i]; if (w>inf) break; dp[i].pb(w); } } } ll count(int n,ll k) { if (n==0) return k==0; k=min(k,(ll)n*(n-1)/2-k); if (k<0) return 0; if (k>=SZ(dp[n])) return inf; else return dp[n][k]; } set<int> s; void gao(int n,ll inv) { for (int i=0;i<n;i++) s.insert(i+1); for (int i=0;i<n;i++) { ll sz=(ll)(n-i-1)*(n-i-2)/2; for (int j=max(0ll,inv-sz);j<n-i;j++) { if (count(n-i-1,inv-j)<k) { k-=count(n-i-1,inv-j); } else { inv-=j; auto it=s.begin(); if (2*j<SZ(s)) { rep(k,0,j) ++it; } else { it=s.end(); rep(k,j,SZ(s)) --it; } printf("%d ",*it); s.erase(it); break; } } } puts(""); } int main() { scanf("%d%lld",&n,&k); if ((ll)n*(n-1)%4!=0) { puts("NIE"); return 0; } init(); if (count(n,(ll)n*(n-1)/4)<k) { puts("NIE"); return 0; } puts("TAK"); ll inv=(ll)n*(n-1)/4; gao(n,inv); } |