#include <bits/stdc++.h>
#define ft first
#define sc second
#define pb push_back
#define FOR(i,n) for(int i=0; i<n; i++)
#define watch(x) cout << (#x) << " == " << (x) << endl
typedef long long ll;
typedef long double ld;
using namespace std;
const ll N = 5e5;
int tab[N+5];
int mmax[N+5];
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int n,k;
cin>>n>>k;
for (int i=1; i<=n; i++){
cin>>tab[i];
}
if (k >= 4){
int x = -1;
for (int i=1; i<=n-1; i++){
if (tab[i] >= tab[i+1]){
x=i;
break;
}
}
if (x == -1){
cout<<"NIE\n";
} else {
cout<<"TAK\n";
if (k >= x+2){
FOR(i,k-1){
cout<<i+1<<" ";
}
} else {
FOR(i,k-4){
cout<<i+1<<" ";
}
cout<<x-1<<" "<<x<<" "<<x+1;
}
}
} else if (k == 3){
for (int i=2; i<=n-1; i++){
if (tab[1] >= tab[i]){
cout<<"TAK\n";
cout<<i-1<<" "<<i;
return 0;
}
}
for (int i=n-1; i>=1; i--){
if (tab[i] >= tab[n]){
cout<<"TAK\n";
cout<<i-1<<" "<<i;
return 0;
}
}
cout<<"NIE";
} else { // n == 2
mmax[n+1]=0;
for (int i=n; i>=1; i--){
mmax[i] = max(mmax[i+1], tab[i]);
}
int mmin=tab[1];
for (int i=1; i<=n-1; i++){
mmin = min(mmin, tab[i]);
if (mmin >= mmax[i+1]){
cout<<"TAK\n";
cout<<i-1<<" "<<i;
return 0;
}
}
cout<<"NIE\n";
}
return 0;
}
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 | #include <bits/stdc++.h> #define ft first #define sc second #define pb push_back #define FOR(i,n) for(int i=0; i<n; i++) #define watch(x) cout << (#x) << " == " << (x) << endl typedef long long ll; typedef long double ld; using namespace std; const ll N = 5e5; int tab[N+5]; int mmax[N+5]; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n,k; cin>>n>>k; for (int i=1; i<=n; i++){ cin>>tab[i]; } if (k >= 4){ int x = -1; for (int i=1; i<=n-1; i++){ if (tab[i] >= tab[i+1]){ x=i; break; } } if (x == -1){ cout<<"NIE\n"; } else { cout<<"TAK\n"; if (k >= x+2){ FOR(i,k-1){ cout<<i+1<<" "; } } else { FOR(i,k-4){ cout<<i+1<<" "; } cout<<x-1<<" "<<x<<" "<<x+1; } } } else if (k == 3){ for (int i=2; i<=n-1; i++){ if (tab[1] >= tab[i]){ cout<<"TAK\n"; cout<<i-1<<" "<<i; return 0; } } for (int i=n-1; i>=1; i--){ if (tab[i] >= tab[n]){ cout<<"TAK\n"; cout<<i-1<<" "<<i; return 0; } } cout<<"NIE"; } else { // n == 2 mmax[n+1]=0; for (int i=n; i>=1; i--){ mmax[i] = max(mmax[i+1], tab[i]); } int mmin=tab[1]; for (int i=1; i<=n-1; i++){ mmin = min(mmin, tab[i]); if (mmin >= mmax[i+1]){ cout<<"TAK\n"; cout<<i-1<<" "<<i; return 0; } } cout<<"NIE\n"; } return 0; } |
English