#include <iostream> #include <list> using namespace std; long long** table; long long* table2; long long int* wyniki; long long kolejna; list<long long int> mylist; long long int fun(long long int n,long long int k) { if(table[n][k] != -1) return table[n][k]; if(k == 0) return 1; if(n == 0) return 0; if(k == 1) return n; long long res = 0; for (long long int i = 0; i <= k && i <= n; i++) res += fun(n-1, k-i); table[n][k] = res; return res; } void make_tables(long long int n) { table = new long long* [n]; for(long long int i = 0; i < n; i++) { table[i] = new long long[50]; for(long long int j = 0; j < 50; j++) { table[i][j] = -1; fun(i, j); } } } int main() { long long int n; long long int k; scanf("%lld", &n); scanf("%lld", &k); if (n%4 >= 2){ printf("NIE\n"); return 0; } wyniki = new long long int[n-1]; for(long long int i = 0; i < n-1;i++) wyniki[i]=0; long long int sum = 0; long long int i = 0; long long int final_res = (n*(n-1))/4; while (sum <= final_res) sum += ++i; make_tables(i); long long int limit = i - 1; sum -= i; wyniki[limit] = final_res-sum; kolejna = wyniki[limit]; sum = 0; long long int prev = 0; i = 0; while (sum < k && sum >= 0) { // Tutaj wstaw naprawianie tabeli i zerowanie "i"; warunek stopu. if (wyniki[limit] == limit + 2){ for (long long int it = 0; it < n - 1 - limit ;it++) { if(wyniki[limit+it] == limit + 2 + it) { if(limit + it == n-2) { printf("NIE\n"); return 0; } wyniki[limit+it+1] += 1; wyniki[limit + it] = 0; continue; } break; } long long tmp = 0; for(long long int it = n - 2; it > limit; it--) tmp += wyniki[it]; wyniki[limit] = kolejna - tmp; if (wyniki[limit] < 0){ i = -wyniki[limit]; wyniki[limit] = 0; } else i = 0; } sum += fun(limit, i); prev = fun(limit, i); i += 1; wyniki[limit] += 1; } sum -= prev; wyniki[limit] -= 1; i -= 1; prev = 0; kolejna = i; long long kolejna2 = kolejna; for(long long int i = limit-1; i >= 0; i--) { if(i+1 > kolejna) { wyniki[i] = i + 1 - kolejna; kolejna = 0; } else { kolejna = kolejna - (i+1); wyniki[i] = 0; } while(sum < k) { if (kolejna == -1) { printf("NIE\n"); return 0; } sum += fun(i, kolejna); prev = fun(i, kolejna); wyniki[i] += 1; kolejna += 1; } wyniki[i] -= 1; kolejna -= 1; sum -= prev; } if (kolejna != 0) { printf("NIE\n"); return 0; } list<long long int>::iterator it; long long int a = 0; for(long long int i = 1; i <= n+1; i++) mylist.push_back(i); it = mylist.begin(); printf("TAK\n"); for(long long int i = n - 2; i >= 0; i--) { advance(it, wyniki[i] - a); a = wyniki[i]; printf("%lld ", *it); it = mylist.erase(it); } advance(it, -a); printf("%lld\n", *it); // Tutaj wstaw dorabianie reszty tabeli //Tutaj wstaw zamiane na permutacje 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 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 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 | #include <iostream> #include <list> using namespace std; long long** table; long long* table2; long long int* wyniki; long long kolejna; list<long long int> mylist; long long int fun(long long int n,long long int k) { if(table[n][k] != -1) return table[n][k]; if(k == 0) return 1; if(n == 0) return 0; if(k == 1) return n; long long res = 0; for (long long int i = 0; i <= k && i <= n; i++) res += fun(n-1, k-i); table[n][k] = res; return res; } void make_tables(long long int n) { table = new long long* [n]; for(long long int i = 0; i < n; i++) { table[i] = new long long[50]; for(long long int j = 0; j < 50; j++) { table[i][j] = -1; fun(i, j); } } } int main() { long long int n; long long int k; scanf("%lld", &n); scanf("%lld", &k); if (n%4 >= 2){ printf("NIE\n"); return 0; } wyniki = new long long int[n-1]; for(long long int i = 0; i < n-1;i++) wyniki[i]=0; long long int sum = 0; long long int i = 0; long long int final_res = (n*(n-1))/4; while (sum <= final_res) sum += ++i; make_tables(i); long long int limit = i - 1; sum -= i; wyniki[limit] = final_res-sum; kolejna = wyniki[limit]; sum = 0; long long int prev = 0; i = 0; while (sum < k && sum >= 0) { // Tutaj wstaw naprawianie tabeli i zerowanie "i"; warunek stopu. if (wyniki[limit] == limit + 2){ for (long long int it = 0; it < n - 1 - limit ;it++) { if(wyniki[limit+it] == limit + 2 + it) { if(limit + it == n-2) { printf("NIE\n"); return 0; } wyniki[limit+it+1] += 1; wyniki[limit + it] = 0; continue; } break; } long long tmp = 0; for(long long int it = n - 2; it > limit; it--) tmp += wyniki[it]; wyniki[limit] = kolejna - tmp; if (wyniki[limit] < 0){ i = -wyniki[limit]; wyniki[limit] = 0; } else i = 0; } sum += fun(limit, i); prev = fun(limit, i); i += 1; wyniki[limit] += 1; } sum -= prev; wyniki[limit] -= 1; i -= 1; prev = 0; kolejna = i; long long kolejna2 = kolejna; for(long long int i = limit-1; i >= 0; i--) { if(i+1 > kolejna) { wyniki[i] = i + 1 - kolejna; kolejna = 0; } else { kolejna = kolejna - (i+1); wyniki[i] = 0; } while(sum < k) { if (kolejna == -1) { printf("NIE\n"); return 0; } sum += fun(i, kolejna); prev = fun(i, kolejna); wyniki[i] += 1; kolejna += 1; } wyniki[i] -= 1; kolejna -= 1; sum -= prev; } if (kolejna != 0) { printf("NIE\n"); return 0; } list<long long int>::iterator it; long long int a = 0; for(long long int i = 1; i <= n+1; i++) mylist.push_back(i); it = mylist.begin(); printf("TAK\n"); for(long long int i = n - 2; i >= 0; i--) { advance(it, wyniki[i] - a); a = wyniki[i]; printf("%lld ", *it); it = mylist.erase(it); } advance(it, -a); printf("%lld\n", *it); // Tutaj wstaw dorabianie reszty tabeli //Tutaj wstaw zamiane na permutacje return 0; } |