#include<bits/stdc++.h> using namespace std; const int N = 1e6 + 5; int n, k, t[N]; void wypisz(int x){ int dowyp = k; if(x == 1){ for(int i = 1 ; i < k ; i++){ cout << i << " "; } } else if(x + 1 == n){ for(int i = 1 ; i < k - 2 ; i++) cout << i <<" " ; cout << n - 2 <<" " << n - 1<<" "; } else{ int ile_musimy = k - 1; ile_musimy -= 3; for(int i = 1 ; i < x - 1 ; i++){ if(ile_musimy > 0){ cout << i <<" "; ile_musimy--; } } cout << x - 1 <<" " << x <<" " << x+1<<" "; for(int i = x + 2 ; i < n ; i++){ if(ile_musimy > 0){ cout << i <<" "; ile_musimy--; } } } } int main(){ ios_base::sync_with_stdio(0); cin.tie(NULL); cin >> n >> k; for(int i = 1 ; i <= n ; i++){ cin >> t[i]; } //k = 2 if(k == 2){ int ob_min = t[1]; multiset<int> s; for(int i = 2 ; i <= n ; i++) s.insert(t[i]); for(int i = 1 ; i < n ; i++){ int ob_max = *s.rbegin(); if(ob_min >= ob_max){ cout <<"TAK\n"; cout << i<<"\n"; return 0; } else{ ob_min = min(ob_min, t[i+1]); s.erase(s.lower_bound(t[i+1])); } } cout <<"NIE\n"; } //k > 3 else if(k > 3){ for(int i = n; i > 1 ; i--){ if(t[i] <= t[i - 1]){ cout <<"TAK\n"; //[1...i-2] [i-1, i-1] [i, i] [i+1, n] (potrzeba k - mamy 2) wypisz(i-1); return 0; } } cout <<"NIE\n"; } else{ //k = 3; //czy min jest na poczotku? jesli nie to git for(int i = 2 ; i <= n ; i++){ if(t[i] <= t[1]){ cout<<"TAK\n"; if(i < n) cout << i - 1 <<" " << i<<"\n"; else cout << 1 << " " << i - 1 <<"\n"; return 0; } } //czy maks jest na koncu? jesli nie to git [1 ... i-1] [i] [i+1 ... n] int maks = t[1]; for(int i = 1 ; i <= n ; i++) maks = max(t[i],maks); for(int i = 2 ; i < n ; i++){ if(t[i] >= maks){ cout <<"TAK\n"; cout << i - 1 <<" " << i<<"\n"; return 0; } } cout <<"NIE\n"; } }
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 | #include<bits/stdc++.h> using namespace std; const int N = 1e6 + 5; int n, k, t[N]; void wypisz(int x){ int dowyp = k; if(x == 1){ for(int i = 1 ; i < k ; i++){ cout << i << " "; } } else if(x + 1 == n){ for(int i = 1 ; i < k - 2 ; i++) cout << i <<" " ; cout << n - 2 <<" " << n - 1<<" "; } else{ int ile_musimy = k - 1; ile_musimy -= 3; for(int i = 1 ; i < x - 1 ; i++){ if(ile_musimy > 0){ cout << i <<" "; ile_musimy--; } } cout << x - 1 <<" " << x <<" " << x+1<<" "; for(int i = x + 2 ; i < n ; i++){ if(ile_musimy > 0){ cout << i <<" "; ile_musimy--; } } } } int main(){ ios_base::sync_with_stdio(0); cin.tie(NULL); cin >> n >> k; for(int i = 1 ; i <= n ; i++){ cin >> t[i]; } //k = 2 if(k == 2){ int ob_min = t[1]; multiset<int> s; for(int i = 2 ; i <= n ; i++) s.insert(t[i]); for(int i = 1 ; i < n ; i++){ int ob_max = *s.rbegin(); if(ob_min >= ob_max){ cout <<"TAK\n"; cout << i<<"\n"; return 0; } else{ ob_min = min(ob_min, t[i+1]); s.erase(s.lower_bound(t[i+1])); } } cout <<"NIE\n"; } //k > 3 else if(k > 3){ for(int i = n; i > 1 ; i--){ if(t[i] <= t[i - 1]){ cout <<"TAK\n"; //[1...i-2] [i-1, i-1] [i, i] [i+1, n] (potrzeba k - mamy 2) wypisz(i-1); return 0; } } cout <<"NIE\n"; } else{ //k = 3; //czy min jest na poczotku? jesli nie to git for(int i = 2 ; i <= n ; i++){ if(t[i] <= t[1]){ cout<<"TAK\n"; if(i < n) cout << i - 1 <<" " << i<<"\n"; else cout << 1 << " " << i - 1 <<"\n"; return 0; } } //czy maks jest na koncu? jesli nie to git [1 ... i-1] [i] [i+1 ... n] int maks = t[1]; for(int i = 1 ; i <= n ; i++) maks = max(t[i],maks); for(int i = 2 ; i < n ; i++){ if(t[i] >= maks){ cout <<"TAK\n"; cout << i - 1 <<" " << i<<"\n"; return 0; } } cout <<"NIE\n"; } } |