#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
int n, k;
cin >> n >> k;
vector<int> v(n), preflmin(n), prefrmax(n);
for (int i = 0; i < n; i++) {
cin >> v[i];
preflmin[i] = v[i];
prefrmax[i] = v[i];
}
for (int i = 1; i < n; i++){
preflmin[i] = min(preflmin[i], preflmin[i - 1]);
}
for (int i = n - 2; i >= 0; i--) {
prefrmax[i] = max(prefrmax[i], prefrmax[i + 1]);
}
if (k == 2) {
for (int i = 0; i < n - 1; i++) {
if (preflmin[i] >= prefrmax[i + 1]) {
cout << "TAK" << endl;
cout << i + 1 << endl;
return 0;
}
}
cout << "NIE" << endl;
}
else if (k == 3) {
for (int i = 1; i < n - 1; i++) {
if (preflmin[i - 1] >= v[i] || v[i] >= prefrmax[i + 1]) {
cout << "TAK" << endl;
cout << i << ' ' << i + 1 << endl;
return 0;
}
}
cout << "NIE" << endl;
}
else {
for (int i = 0; i < n - 1; i++) {
if (v[i] >= v[i + 1]) {
cout << "TAK" << endl;
set<int> s;
s.insert(i);
s.insert(i + 1);
if (i > 0) s.insert(i - 1);
s.insert(n - 1);
for (int j = 0; j < n; j++) {
if (s.size() < k) s.insert(j);
}
set<int>::iterator it;
for (it = s.begin(); it != s.end() && k > 1; it++) {
cout << (*it) + 1 << ' ';
k--;
}
return 0;
}
}
cout << "NIE" << endl;
}
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 | #include<iostream> #include<vector> #include<set> #include<algorithm> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); int n, k; cin >> n >> k; vector<int> v(n), preflmin(n), prefrmax(n); for (int i = 0; i < n; i++) { cin >> v[i]; preflmin[i] = v[i]; prefrmax[i] = v[i]; } for (int i = 1; i < n; i++){ preflmin[i] = min(preflmin[i], preflmin[i - 1]); } for (int i = n - 2; i >= 0; i--) { prefrmax[i] = max(prefrmax[i], prefrmax[i + 1]); } if (k == 2) { for (int i = 0; i < n - 1; i++) { if (preflmin[i] >= prefrmax[i + 1]) { cout << "TAK" << endl; cout << i + 1 << endl; return 0; } } cout << "NIE" << endl; } else if (k == 3) { for (int i = 1; i < n - 1; i++) { if (preflmin[i - 1] >= v[i] || v[i] >= prefrmax[i + 1]) { cout << "TAK" << endl; cout << i << ' ' << i + 1 << endl; return 0; } } cout << "NIE" << endl; } else { for (int i = 0; i < n - 1; i++) { if (v[i] >= v[i + 1]) { cout << "TAK" << endl; set<int> s; s.insert(i); s.insert(i + 1); if (i > 0) s.insert(i - 1); s.insert(n - 1); for (int j = 0; j < n; j++) { if (s.size() < k) s.insert(j); } set<int>::iterator it; for (it = s.begin(); it != s.end() && k > 1; it++) { cout << (*it) + 1 << ' '; k--; } return 0; } } cout << "NIE" << endl; } return 0; } |
English