#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"; } } |
English