#include <iostream>
#define nl '\n'
using namespace std;
const int MAX_N = 500007;
int n,k;
int przychod[MAX_N];
int maxes[MAX_N];
int mins[MAX_N];
int used=-1;
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cin>>n>>k;
for(int i=0; i<n; ++i){
cin>>przychod[i];
}
if(k==2){
mins[0]=przychod[0];
for(int i=1; i<n; ++i){
mins[i]=min(mins[i-1],przychod[i]);
}
maxes[n-1]=przychod[n-1];
for(int i=n-2; i>=0; --i){
maxes[i]=max(maxes[i+1],przychod[i]);
}
for(int i=0; i<n-1; ++i){
if(maxes[i+1]<=mins[i]){
cout<<"TAK"<<nl<<i+1<<nl;
return 0;
}
}
cout<<"NIE"<<nl;
return 0;
}
else if(k==3){
for(int i=1; i<n-1; ++i){
if(przychod[i]<=przychod[0]){
cout<<"TAK"<<nl;
cout<<i<<' '<<i+1<<nl;
return 0;
}
}
for(int i=n-2; i>0; --i){
if(przychod[i]>=przychod[n-1]){
cout<<"TAK"<<nl;
cout<<i<<' '<<i+1<<nl;
return 0;
}
}
cout<<"NIE"<<nl;
return 0;
}
else if(k>3){
for(int i=1; i<n; ++i){
if(przychod[i]<=przychod[i-1]){
used = i;
break;
}
}
if(used==-1){
cout<<"NIE"<<nl;
return 0;
}
cout<<"TAK"<<nl;
if(used>k-2){
for(int i=1; i<=k-3; ++i){
cout<<i<<' ';
}
cout<<used-1<<' '<<used<<' '<<used+1<<nl;
return 0;
}
else{
for(int i=1; i<=k-1; ++i){
cout<<i<<' ';
}
cout<<nl;
return 0;
}
}
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 | #include <iostream> #define nl '\n' using namespace std; const int MAX_N = 500007; int n,k; int przychod[MAX_N]; int maxes[MAX_N]; int mins[MAX_N]; int used=-1; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin>>n>>k; for(int i=0; i<n; ++i){ cin>>przychod[i]; } if(k==2){ mins[0]=przychod[0]; for(int i=1; i<n; ++i){ mins[i]=min(mins[i-1],przychod[i]); } maxes[n-1]=przychod[n-1]; for(int i=n-2; i>=0; --i){ maxes[i]=max(maxes[i+1],przychod[i]); } for(int i=0; i<n-1; ++i){ if(maxes[i+1]<=mins[i]){ cout<<"TAK"<<nl<<i+1<<nl; return 0; } } cout<<"NIE"<<nl; return 0; } else if(k==3){ for(int i=1; i<n-1; ++i){ if(przychod[i]<=przychod[0]){ cout<<"TAK"<<nl; cout<<i<<' '<<i+1<<nl; return 0; } } for(int i=n-2; i>0; --i){ if(przychod[i]>=przychod[n-1]){ cout<<"TAK"<<nl; cout<<i<<' '<<i+1<<nl; return 0; } } cout<<"NIE"<<nl; return 0; } else if(k>3){ for(int i=1; i<n; ++i){ if(przychod[i]<=przychod[i-1]){ used = i; break; } } if(used==-1){ cout<<"NIE"<<nl; return 0; } cout<<"TAK"<<nl; if(used>k-2){ for(int i=1; i<=k-3; ++i){ cout<<i<<' '; } cout<<used-1<<' '<<used<<' '<<used+1<<nl; return 0; } else{ for(int i=1; i<=k-1; ++i){ cout<<i<<' '; } cout<<nl; return 0; } } return 0; } |
English