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