//Mateusz Kruk
#include <bits/stdc++.h>
using namespace std;
int main(){
int n,k;
cin>>n>>k;
vector<int>A;
for(int i=0;i<n;i++){
int a;
cin>>a;
A.push_back(a);
}
int j=1;
while(A[j]>A[0]&&n-j>=k){
j++;
}
vector<int>ans;
if(A[j]<=A[0]){
if(k!=2){
cout<<"TAK"<<'\n';
cout<<j<<" ";
int x=k-2;
for(int h=j+1;h<j+1+x;h++){
cout<<h<<" ";
}
cout<<'\n';
}
else{
bool e=true;
int h=j+1;
while(j<n){
if(A[j]>A[0]){
e=false;
}
j++;
}
if(e==true){
cout<<"TAK"<<'\n';
cout<<h-1<<'\n';
}
else{
cout<<"NIE"<<'\n';
}
}
}
else{
ans.push_back(j);
bool e=false;
for(int i=j+1;i<n;i++){
ans.push_back(i);
if(A[i-1]>A[i]){
e=true;
}
}
if(e==true){
cout<<"TAK"<<'\n';
for(int i=0;i<ans.size();i++){
cout<<ans[i]<<" ";
}
cout<<'\n';
}
else{
cout<<"NIE"<<'\n';
}
}
}
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 | //Mateusz Kruk #include <bits/stdc++.h> using namespace std; int main(){ int n,k; cin>>n>>k; vector<int>A; for(int i=0;i<n;i++){ int a; cin>>a; A.push_back(a); } int j=1; while(A[j]>A[0]&&n-j>=k){ j++; } vector<int>ans; if(A[j]<=A[0]){ if(k!=2){ cout<<"TAK"<<'\n'; cout<<j<<" "; int x=k-2; for(int h=j+1;h<j+1+x;h++){ cout<<h<<" "; } cout<<'\n'; } else{ bool e=true; int h=j+1; while(j<n){ if(A[j]>A[0]){ e=false; } j++; } if(e==true){ cout<<"TAK"<<'\n'; cout<<h-1<<'\n'; } else{ cout<<"NIE"<<'\n'; } } } else{ ans.push_back(j); bool e=false; for(int i=j+1;i<n;i++){ ans.push_back(i); if(A[i-1]>A[i]){ e=true; } } if(e==true){ cout<<"TAK"<<'\n'; for(int i=0;i<ans.size();i++){ cout<<ans[i]<<" "; } cout<<'\n'; } else{ cout<<"NIE"<<'\n'; } } } |
English