// Marcin Knapik
#pragma GCC optimize ("O3")
#include<bits/stdc++.h>
using namespace std;
#define FOR(i, n) for (int i = 0; i < n; i++)
#define f first
#define s second
#define pb push_back
#define all(s) s.begin(), s.end()
#define sz(s) (int)s.size()
using ll = long long;
using vi = vector<int>;
void solve () {
int n, k;
cin >> n >> k;
vi tab(n);
for(auto & u : tab){
cin >> u;
}
vi pref(n);
vi suf(n);
pref[0] = tab[0];
for(int i = 1; i < n; i++){
pref[i] = min(tab[i], pref[i - 1]);
}
for (int i = n - 2; i >= 0; i--) {
suf[i] = max(suf[i + 1], tab[i + 1]);
}
set<int> przedzialki;
if(k == 2){
for(int i = 0; i < n - 1; i++){
if(pref[i] >= suf[i]){
przedzialki.insert(i + 1);
break;
}
}
} else if (k == 3){
for(int i = 0; i < n - 1; i++){
if(tab[i] >= suf[i]){
przedzialki.insert(i + 1);
if(i > 0) {
przedzialki.insert(i);
}
break;
}
}
if(sz(przedzialki) == 0){
for(int i = 1; i < n; i++){
if(tab[i] <= pref[i - 1]){
przedzialki.insert(i);
if(i < n - 1){
przedzialki.insert(i + 1);
}
break;
}
}
}
} else {
for(int i = 0; i < n - 1; i++){
if(tab[i] >= tab[i + 1]){
if(i > 0) {
przedzialki.insert(i);
}
przedzialki.insert(i + 1);
if(i < n - 2){
przedzialki.insert(i + 2);
}
break;
}
}
}
if(sz(przedzialki) == 0){
cout << "NIE" << '\n';
} else {
cout << "TAK\n";
for(int i = 1; i < n; i ++){
if(sz(przedzialki) < k - 1){
przedzialki.insert(i);
}
}
for(auto & u : przedzialki){
cout << u << ' ';
}
cout << '\n';
}
}
int main () {
ios::sync_with_stdio(0);
cin.tie(0);
int tests = 1;
// cin >> tests;
for (int test = 1; test <= tests; test++) {
solve();
}
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 | // Marcin Knapik #pragma GCC optimize ("O3") #include<bits/stdc++.h> using namespace std; #define FOR(i, n) for (int i = 0; i < n; i++) #define f first #define s second #define pb push_back #define all(s) s.begin(), s.end() #define sz(s) (int)s.size() using ll = long long; using vi = vector<int>; void solve () { int n, k; cin >> n >> k; vi tab(n); for(auto & u : tab){ cin >> u; } vi pref(n); vi suf(n); pref[0] = tab[0]; for(int i = 1; i < n; i++){ pref[i] = min(tab[i], pref[i - 1]); } for (int i = n - 2; i >= 0; i--) { suf[i] = max(suf[i + 1], tab[i + 1]); } set<int> przedzialki; if(k == 2){ for(int i = 0; i < n - 1; i++){ if(pref[i] >= suf[i]){ przedzialki.insert(i + 1); break; } } } else if (k == 3){ for(int i = 0; i < n - 1; i++){ if(tab[i] >= suf[i]){ przedzialki.insert(i + 1); if(i > 0) { przedzialki.insert(i); } break; } } if(sz(przedzialki) == 0){ for(int i = 1; i < n; i++){ if(tab[i] <= pref[i - 1]){ przedzialki.insert(i); if(i < n - 1){ przedzialki.insert(i + 1); } break; } } } } else { for(int i = 0; i < n - 1; i++){ if(tab[i] >= tab[i + 1]){ if(i > 0) { przedzialki.insert(i); } przedzialki.insert(i + 1); if(i < n - 2){ przedzialki.insert(i + 2); } break; } } } if(sz(przedzialki) == 0){ cout << "NIE" << '\n'; } else { cout << "TAK\n"; for(int i = 1; i < n; i ++){ if(sz(przedzialki) < k - 1){ przedzialki.insert(i); } } for(auto & u : przedzialki){ cout << u << ' '; } cout << '\n'; } } int main () { ios::sync_with_stdio(0); cin.tie(0); int tests = 1; // cin >> tests; for (int test = 1; test <= tests; test++) { solve(); } return 0; } |
English