//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'; } } } |