#include <bits/stdc++.h> using namespace std; const int MAXN = 500010; int A[MAXN]; int premin[MAXN]; int sufmax[MAXN]; void yes(vector<int> v) { puts("TAK"); for (auto x : v) printf("%d ", x); puts(""); exit(0); } void no() { puts("NIE"); exit(0); } int main() { int n, k; scanf("%d%d", &n, &k); for (int i = 1; i <= n; i++) { scanf("%d", A + i); } premin[1] = A[1]; for (int i = 2; i <= n; i++) { premin[i] = min(premin[i - 1], A[i]); } sufmax[n] = A[n]; for (int i = n - 1; i >= 1; i--) { sufmax[i] = max(sufmax[i + 1], A[i]); } if (k == 2) { for (int i = 1; i < n; i++) { if (premin[i] >= sufmax[i + 1]) { yes({ i }); } } no(); } if (k == 3) { if (A[1] >= A[n]) { yes({ 1, n - 1 }); } for (int i = 2; i < n; i++) { if (A[1] >= A[i]) { yes({ i - 1, i }); } } for (int i = n - 1; i > 1; i--) { if (A[i] >= A[n]) { yes({ i - 1, i }); } } no(); } if (k >= 4) { if (A[1] >= A[2]) { vector<int> out; for (int j = 1; j < k; j++) { out.push_back(j); } yes(out); } for (int i = 1; i < n; i++) { if (A[i] >= A[i + 1]) { vector<int> out; for (int j = 1; j < k - (n - i + 1); j++) { out.push_back(j); } out.push_back(i - 1); for (int j = i; j < min(n, i + k - 2); j++) { out.push_back(j); } yes(out); } } no(); } }
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 | #include <bits/stdc++.h> using namespace std; const int MAXN = 500010; int A[MAXN]; int premin[MAXN]; int sufmax[MAXN]; void yes(vector<int> v) { puts("TAK"); for (auto x : v) printf("%d ", x); puts(""); exit(0); } void no() { puts("NIE"); exit(0); } int main() { int n, k; scanf("%d%d", &n, &k); for (int i = 1; i <= n; i++) { scanf("%d", A + i); } premin[1] = A[1]; for (int i = 2; i <= n; i++) { premin[i] = min(premin[i - 1], A[i]); } sufmax[n] = A[n]; for (int i = n - 1; i >= 1; i--) { sufmax[i] = max(sufmax[i + 1], A[i]); } if (k == 2) { for (int i = 1; i < n; i++) { if (premin[i] >= sufmax[i + 1]) { yes({ i }); } } no(); } if (k == 3) { if (A[1] >= A[n]) { yes({ 1, n - 1 }); } for (int i = 2; i < n; i++) { if (A[1] >= A[i]) { yes({ i - 1, i }); } } for (int i = n - 1; i > 1; i--) { if (A[i] >= A[n]) { yes({ i - 1, i }); } } no(); } if (k >= 4) { if (A[1] >= A[2]) { vector<int> out; for (int j = 1; j < k; j++) { out.push_back(j); } yes(out); } for (int i = 1; i < n; i++) { if (A[i] >= A[i + 1]) { vector<int> out; for (int j = 1; j < k - (n - i + 1); j++) { out.push_back(j); } out.push_back(i - 1); for (int j = i; j < min(n, i + k - 2); j++) { out.push_back(j); } yes(out); } } no(); } } |