#include <cstdio> #include <utility> #include <cstring> #include <math.h> bool genPerm(int a[], const int &n) { int x = -1; for(int i = 0; i < n - 1; ++i) { if(a[i] < a[i+1]){ x = i; } } if(x == -1){ return false; } int y = -1; for(int j = 0; j < n; ++j) { if(a[x] < a[j]){ y = j; } } std::swap(a[x], a[y]); for(int i = x + 1; i <= (n-x)/2 + x; ++i) { std::swap(a[i], a[n + x - i]); } return true; } bool czyStablina(int a[], const int &n) { int liczInw = 0; for (int i = 0; i < n; ++i) { for (int j = i; j < n; ++j) { if (a[i] > a[j]) ++liczInw; } } int liczInw2 = 0; for (int i = 0; i < n; ++i) { for (int j = i; j < n; ++j) { if (a[i] < a[j]) ++liczInw2; } } return (liczInw == liczInw2); } int main() { int n; unsigned long long k; scanf("%d %llu", &n, &k); if(k >= pow(n/2 + 1, n) ){ printf("NIE\n"); return 0; } int a[n]; for(int i = 0; i < n; ++i){ a[i] = i+1; } unsigned long long idPermutacji = 0; while(genPerm(a, n)) { if(czyStablina(a, n)){ ++idPermutacji; if(idPermutacji == k){ printf("TAK\n"); for(int i = 0; i < n; ++i){ printf("%d ", a[i]); } printf("\n"); return 0; } } } printf("NIE\n"); 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 93 94 95 96 | #include <cstdio> #include <utility> #include <cstring> #include <math.h> bool genPerm(int a[], const int &n) { int x = -1; for(int i = 0; i < n - 1; ++i) { if(a[i] < a[i+1]){ x = i; } } if(x == -1){ return false; } int y = -1; for(int j = 0; j < n; ++j) { if(a[x] < a[j]){ y = j; } } std::swap(a[x], a[y]); for(int i = x + 1; i <= (n-x)/2 + x; ++i) { std::swap(a[i], a[n + x - i]); } return true; } bool czyStablina(int a[], const int &n) { int liczInw = 0; for (int i = 0; i < n; ++i) { for (int j = i; j < n; ++j) { if (a[i] > a[j]) ++liczInw; } } int liczInw2 = 0; for (int i = 0; i < n; ++i) { for (int j = i; j < n; ++j) { if (a[i] < a[j]) ++liczInw2; } } return (liczInw == liczInw2); } int main() { int n; unsigned long long k; scanf("%d %llu", &n, &k); if(k >= pow(n/2 + 1, n) ){ printf("NIE\n"); return 0; } int a[n]; for(int i = 0; i < n; ++i){ a[i] = i+1; } unsigned long long idPermutacji = 0; while(genPerm(a, n)) { if(czyStablina(a, n)){ ++idPermutacji; if(idPermutacji == k){ printf("TAK\n"); for(int i = 0; i < n; ++i){ printf("%d ", a[i]); } printf("\n"); return 0; } } } printf("NIE\n"); return 0; } |