#include<bits/stdc++.h>
using namespace std;
bool odw[500002],bul;
int n,k,maks,ile,gdzie;
int tab[500002],premin[500002],sufmax[500002];
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
cin>>n>>k;
premin[0]=1000000001;
for(int i=1;i<=n;++i){
cin>>tab[i];
premin[i]=min(tab[i],premin[i-1]);
sufmax[i]=tab[i];
}
if(k==2){
for(int i=n-1;i;--i){
if(sufmax[i]<sufmax[i+1])sufmax[i]=sufmax[i+1];
}
for(int i=1;i<n;++i){
if(premin[i]>=sufmax[i+1]){
cout<<"TAK"<<"\n";
cout<<i;
return 0;
}
}
cout<<"NIE";
}
else if(k==3){
for(int i=n-1;i;--i){
if(sufmax[i]<sufmax[i+1])sufmax[i]=sufmax[i+1];
}
for(int i=2;i<n;++i){
if(tab[i]<=premin[i-1]||tab[i]>=sufmax[i+1]){
cout<<"TAK"<<"\n";
cout<<i-1<<" "<<i;
return 0;
}
}
cout<<"NIE";
}
else{
for(int i=1;i<n;++i){
if(tab[i]>=tab[i+1]){
bul=1;
k-=2;
odw[i]=1;odw[i+1]=1;
if(i-1){
--k;
odw[i-1]=1;
}
i=n;
}
}
if(!bul){
cout<<"NIE";
return 0;
}
--k;
cout<<"TAK"<<"\n";
for(int i=1;i<n;++i){
if(k||odw[i]){
cout<<i<<" ";
if(!odw[i]){
--k;
}
}
}
}
}
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 | #include<bits/stdc++.h> using namespace std; bool odw[500002],bul; int n,k,maks,ile,gdzie; int tab[500002],premin[500002],sufmax[500002]; int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cin>>n>>k; premin[0]=1000000001; for(int i=1;i<=n;++i){ cin>>tab[i]; premin[i]=min(tab[i],premin[i-1]); sufmax[i]=tab[i]; } if(k==2){ for(int i=n-1;i;--i){ if(sufmax[i]<sufmax[i+1])sufmax[i]=sufmax[i+1]; } for(int i=1;i<n;++i){ if(premin[i]>=sufmax[i+1]){ cout<<"TAK"<<"\n"; cout<<i; return 0; } } cout<<"NIE"; } else if(k==3){ for(int i=n-1;i;--i){ if(sufmax[i]<sufmax[i+1])sufmax[i]=sufmax[i+1]; } for(int i=2;i<n;++i){ if(tab[i]<=premin[i-1]||tab[i]>=sufmax[i+1]){ cout<<"TAK"<<"\n"; cout<<i-1<<" "<<i; return 0; } } cout<<"NIE"; } else{ for(int i=1;i<n;++i){ if(tab[i]>=tab[i+1]){ bul=1; k-=2; odw[i]=1;odw[i+1]=1; if(i-1){ --k; odw[i-1]=1; } i=n; } } if(!bul){ cout<<"NIE"; return 0; } --k; cout<<"TAK"<<"\n"; for(int i=1;i<n;++i){ if(k||odw[i]){ cout<<i<<" "; if(!odw[i]){ --k; } } } } } |
English