// 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; } |