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