#include <bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n,k,najm,najw,a=0,poz_a,b=0,poz_b; cin>>n>>k; int T[n+1]; int maksowe[n+1]; int odp[n+1]; for(int i=1;i<=n;i++){ cin>>T[i]; } for(int i=1;i<=n;i++){ odp[i]=0; } if(k==2){ maksowe[n]=T[n]; for(int i=n-1;i>0;i--){ maksowe[i]=max(maksowe[i+1], T[i]); } /* for(int i=0;i<n;i++){ cout<<maksowe[i]<<" "; } cout<<endl; */ najm=T[1]; for(int i=1;i<n;i++){ najm=min(najm, T[i]); if(najm>=maksowe[i+1]){ a=i; break; } } if(a!=0){ cout<<"TAK"<<endl; cout<<a; }else{ cout<<"NIE"; } }else if(k==3){ najm=T[1]; a=1; for(int i=2;i<=n;i++){ if(T[i]<=najm){ najm=T[i]; a=i; } } najw=T[n]; b=n; for(int i=n-1;i>=1;i--){ if(T[i]>=najw){ najw=T[i]; b=i; } } if(a!=1){ cout<<"TAK"<<endl; if(a!=n){ cout<<a-1<<" "<<a; }else{ cout<<1<<" "<<a-1; } }else if(b!=n){ cout<<"TAK"<<endl; if(b!=1){ cout<<b-1<<" "<<b; }else{ cout<<1<<" "<<n-1; } }else{ cout<<"NIE"; } }else{ najw=T[1]; for(int i=2;i<=n;i++){ if(T[i]<=T[i-1]){ a=i-1; b=i; break; } } if(a==0){ cout<<"NIE"; }else{ cout<<"TAK"<<endl; if(a!=1){ odp[a-1]=1; k-=1; } odp[a]=1; k-=1; if(b!=n){ odp[b]=1; k-=1; } for(int i=1;i<=n;i++){ if(k==1){ break; } if(odp[i]==0){ odp[i]=1; k-=1; } } for(int i=1;i<=n;i++){ if(odp[i]==1){ cout<<i<<" "; } } } } }
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 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 | #include <bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n,k,najm,najw,a=0,poz_a,b=0,poz_b; cin>>n>>k; int T[n+1]; int maksowe[n+1]; int odp[n+1]; for(int i=1;i<=n;i++){ cin>>T[i]; } for(int i=1;i<=n;i++){ odp[i]=0; } if(k==2){ maksowe[n]=T[n]; for(int i=n-1;i>0;i--){ maksowe[i]=max(maksowe[i+1], T[i]); } /* for(int i=0;i<n;i++){ cout<<maksowe[i]<<" "; } cout<<endl; */ najm=T[1]; for(int i=1;i<n;i++){ najm=min(najm, T[i]); if(najm>=maksowe[i+1]){ a=i; break; } } if(a!=0){ cout<<"TAK"<<endl; cout<<a; }else{ cout<<"NIE"; } }else if(k==3){ najm=T[1]; a=1; for(int i=2;i<=n;i++){ if(T[i]<=najm){ najm=T[i]; a=i; } } najw=T[n]; b=n; for(int i=n-1;i>=1;i--){ if(T[i]>=najw){ najw=T[i]; b=i; } } if(a!=1){ cout<<"TAK"<<endl; if(a!=n){ cout<<a-1<<" "<<a; }else{ cout<<1<<" "<<a-1; } }else if(b!=n){ cout<<"TAK"<<endl; if(b!=1){ cout<<b-1<<" "<<b; }else{ cout<<1<<" "<<n-1; } }else{ cout<<"NIE"; } }else{ najw=T[1]; for(int i=2;i<=n;i++){ if(T[i]<=T[i-1]){ a=i-1; b=i; break; } } if(a==0){ cout<<"NIE"; }else{ cout<<"TAK"<<endl; if(a!=1){ odp[a-1]=1; k-=1; } odp[a]=1; k-=1; if(b!=n){ odp[b]=1; k-=1; } for(int i=1;i<=n;i++){ if(k==1){ break; } if(odp[i]==0){ odp[i]=1; k-=1; } } for(int i=1;i<=n;i++){ if(odp[i]==1){ cout<<i<<" "; } } } } } |