#include <bits/stdc++.h> typedef long long int lli; using namespace std; const int INF = 1e9 + 3; int main(){ ios_base::sync_with_stdio(false); cin.tie(NULL); int n, k, mn = INF, mx = -INF, pos_mn = -1, pos_mx = -1; cin >> n >> k; vector <int> a(n); for (int i = 0; i < n; i++){ cin >> a[i]; mn = min(mn, a[i]); mx = max(mx, a[i]); } bool sorted = true, min_not_first = false, max_not_last = false; for (int i = 1; i < n; i++){ sorted &= (a[i] > a[i - 1]); if (a[i] == mn){ pos_mn = i; min_not_first = true; } if (a[i - 1] == mx){ pos_mx = i - 1; max_not_last = true; } } if (sorted || (!min_not_first && !max_not_last && k == 3)){ cout << "NIE\n"; return 0; } vector <int> ans; if (k >= 3){ if (min_not_first){ ans.push_back(pos_mn); if (pos_mn != n - 1){ ans.push_back(pos_mn + 1); k -= 3; } else{ k -= 2; } for (int i = 0; i < n; i++){ if (k && (i != pos_mn - 1) && (i != pos_mn)){ ans.push_back(i + 1); k--; } } } else if (max_not_last){ ans.push_back(pos_mx + 1); if (pos_mx != 0){ ans.push_back(pos_mx); k -= 3; } else{ k -= 2; } for (int i = 0; i < n; i++){ if (k && (i != pos_mx - 1) && (i != pos_mx)){ ans.push_back(i + 1); k--; } } } else{ int where = -1; for (int i = 0; i < n - 1; i++){ if (a[i] >= a[i + 1]){ ans.push_back(i + 1); ans.push_back(i + 2); where = i + 1; break; // this must happen since the sequence isn't sorted } } k -= 3; for (int i = 0; i < n; i++){ if (k && (i != where) && (i != where + 1)){ ans.push_back(i + 1); k--; } } } cout << "TAK\n"; sort(ans.begin(), ans.end()); for (int x : ans){ cout << x << " "; } cout << "\n"; } else{ // k == 2, naive multiset <int> s; for (int i = 0; i < n; i++){ s.insert(a[i]); } mn = INF; for (int i = 0; i < n - 1; i++){ mn = min(mn, a[i]); s.erase(s.find(a[i])); if (mn >= *(s.rbegin())){ cout << "TAK\n"; cout << i + 1 << "\n"; return 0; } } cout << "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 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | #include <bits/stdc++.h> typedef long long int lli; using namespace std; const int INF = 1e9 + 3; int main(){ ios_base::sync_with_stdio(false); cin.tie(NULL); int n, k, mn = INF, mx = -INF, pos_mn = -1, pos_mx = -1; cin >> n >> k; vector <int> a(n); for (int i = 0; i < n; i++){ cin >> a[i]; mn = min(mn, a[i]); mx = max(mx, a[i]); } bool sorted = true, min_not_first = false, max_not_last = false; for (int i = 1; i < n; i++){ sorted &= (a[i] > a[i - 1]); if (a[i] == mn){ pos_mn = i; min_not_first = true; } if (a[i - 1] == mx){ pos_mx = i - 1; max_not_last = true; } } if (sorted || (!min_not_first && !max_not_last && k == 3)){ cout << "NIE\n"; return 0; } vector <int> ans; if (k >= 3){ if (min_not_first){ ans.push_back(pos_mn); if (pos_mn != n - 1){ ans.push_back(pos_mn + 1); k -= 3; } else{ k -= 2; } for (int i = 0; i < n; i++){ if (k && (i != pos_mn - 1) && (i != pos_mn)){ ans.push_back(i + 1); k--; } } } else if (max_not_last){ ans.push_back(pos_mx + 1); if (pos_mx != 0){ ans.push_back(pos_mx); k -= 3; } else{ k -= 2; } for (int i = 0; i < n; i++){ if (k && (i != pos_mx - 1) && (i != pos_mx)){ ans.push_back(i + 1); k--; } } } else{ int where = -1; for (int i = 0; i < n - 1; i++){ if (a[i] >= a[i + 1]){ ans.push_back(i + 1); ans.push_back(i + 2); where = i + 1; break; // this must happen since the sequence isn't sorted } } k -= 3; for (int i = 0; i < n; i++){ if (k && (i != where) && (i != where + 1)){ ans.push_back(i + 1); k--; } } } cout << "TAK\n"; sort(ans.begin(), ans.end()); for (int x : ans){ cout << x << " "; } cout << "\n"; } else{ // k == 2, naive multiset <int> s; for (int i = 0; i < n; i++){ s.insert(a[i]); } mn = INF; for (int i = 0; i < n - 1; i++){ mn = min(mn, a[i]); s.erase(s.find(a[i])); if (mn >= *(s.rbegin())){ cout << "TAK\n"; cout << i + 1 << "\n"; return 0; } } cout << "NIE\n"; } return 0; } |