#include <bits/stdc++.h> using namespace std; // #define DEBUG #ifdef DEBUG #define LOG(s) cerr << s << '\n' const bool debug = true; #else #define LOG(s) const bool debug = false; #endif int main() { ios_base::sync_with_stdio(0); cin.tie(NULL); // Wczytywanie danych: int n, k; cin >> n >> k; int tab[n]; for(int i = 0; i < n; i++) { cin >> tab[i]; } // Przypadek 1: if(k > 3) { int w1 = 1; while(w1 < n && tab[w1] > tab[w1 - 1]) { w1++; } if(w1 == n) { cout << "NIE\n"; return 0; } cout << "TAK\n"; if(w1 == 1) { for(int i = 1; i < k; i++) { cout << i << ' '; } cout << '\n'; return 0; } int i = 0; while(k > 4 && i < w1 - 2) { cout << i + 1 << ' '; i++; k--; } cout << w1 - 1 << ' ' << w1 << ' ' << w1 + 1; i = w1 + 2; k -= 4; while(k > 0) { cout << ' ' << i; i++; k--; } cout << '\n'; return 0; } // Przypadek 2: if(k == 3) { int i = 1; while(i < n - 1 && tab[0] < tab[i] && tab[i] < tab[n - 1]) { i++; } if(i == n - 1) { cout << "NIE\n"; return 0; } if(tab[0] >= tab[i]) { cout << "TAK\n" << i << ' ' << i + 1 << '\n'; return 0; } i = n - 2; while(tab[i] < tab[n - 1]) { i--; } cout << "TAK\n" << i << ' ' << i + 1 << '\n'; return 0; } // Przypadek 3 (k == 2): int maxSuf[n]; maxSuf[n - 1] = tab[n - 1]; for(int i = n - 2; i > -1; i--) { maxSuf[i] = max(tab[i], maxSuf[i + 1]); } int minP = tab[0]; for(int i = 1; i < n; i++) { if(minP >= maxSuf[i]) { cout << "TAK\n" << i << '\n'; return 0; } minP = min(minP, tab[i]); } 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 | #include <bits/stdc++.h> using namespace std; // #define DEBUG #ifdef DEBUG #define LOG(s) cerr << s << '\n' const bool debug = true; #else #define LOG(s) const bool debug = false; #endif int main() { ios_base::sync_with_stdio(0); cin.tie(NULL); // Wczytywanie danych: int n, k; cin >> n >> k; int tab[n]; for(int i = 0; i < n; i++) { cin >> tab[i]; } // Przypadek 1: if(k > 3) { int w1 = 1; while(w1 < n && tab[w1] > tab[w1 - 1]) { w1++; } if(w1 == n) { cout << "NIE\n"; return 0; } cout << "TAK\n"; if(w1 == 1) { for(int i = 1; i < k; i++) { cout << i << ' '; } cout << '\n'; return 0; } int i = 0; while(k > 4 && i < w1 - 2) { cout << i + 1 << ' '; i++; k--; } cout << w1 - 1 << ' ' << w1 << ' ' << w1 + 1; i = w1 + 2; k -= 4; while(k > 0) { cout << ' ' << i; i++; k--; } cout << '\n'; return 0; } // Przypadek 2: if(k == 3) { int i = 1; while(i < n - 1 && tab[0] < tab[i] && tab[i] < tab[n - 1]) { i++; } if(i == n - 1) { cout << "NIE\n"; return 0; } if(tab[0] >= tab[i]) { cout << "TAK\n" << i << ' ' << i + 1 << '\n'; return 0; } i = n - 2; while(tab[i] < tab[n - 1]) { i--; } cout << "TAK\n" << i << ' ' << i + 1 << '\n'; return 0; } // Przypadek 3 (k == 2): int maxSuf[n]; maxSuf[n - 1] = tab[n - 1]; for(int i = n - 2; i > -1; i--) { maxSuf[i] = max(tab[i], maxSuf[i + 1]); } int minP = tab[0]; for(int i = 1; i < n; i++) { if(minP >= maxSuf[i]) { cout << "TAK\n" << i << '\n'; return 0; } minP = min(minP, tab[i]); } cout << "NIE\n"; return 0; } |