#include <bits/stdc++.h>
using namespace std;
#define int int64_t
#define pb push_back
#define st first
#define nd second
#define pii pair<int, int>
#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
int32_t main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n, k;
cin >> n >> k;
vector<int> v(n);
for(int i=0;i<n;i++) cin >> v[i];
if(k == 2) {
vector<int> Max(n);
Max[n-1] = v[n-1];
for(int i=n-2;i>=0;i--) Max[i] = max(Max[i+1], v[i]);
int Min = 1e9;
for(int i=0;i<n-1;i++) {
Min = min(Min, v[i]);
if(Min >= Max[i+1]) {
cout << "TAK\n";
cout << i+1 << "\n";
return 0;
}
}
cout << "NIE\n";
return 0;
}
if(k == 3) {
int Max = *max_element(all(v));
int Min = *min_element(all(v));
for(int i=0;i<n-1;i++) {
if(v[i] == Max) {
cout << "TAK\n";
cout << i << " " << i+1 << "\n";
return 0;
}
}
for(int i=1;i<n;i++) {
if(v[i] == Min) {
cout << "TAK\n";
cout << i << " " << i+1 << "\n";
return 0;
}
}
cout << "NIE\n";
return 0;
}
for(int i=0;i<n-1;i++) {
if(v[i] >= v[i+1]) {
cout << "TAK\n";
int rem = k-4;
for(int j=0;j<i-1&&rem;j++) {
cout << j+1 << " ";
rem--;
}
cout << i << " " << i+1 << " " << i+2 << " ";
for(int j=i+2;j<n&&rem;j++) {
cout << j+1 << " ";
rem--;
}
cout << "\n";
return 0;
}
}
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 67 68 69 70 71 72 73 74 | #include <bits/stdc++.h> using namespace std; #define int int64_t #define pb push_back #define st first #define nd second #define pii pair<int, int> #define all(x) (x).begin(), (x).end() #define sz(x) (int)(x).size() int32_t main() { ios_base::sync_with_stdio(false); cin.tie(NULL); int n, k; cin >> n >> k; vector<int> v(n); for(int i=0;i<n;i++) cin >> v[i]; if(k == 2) { vector<int> Max(n); Max[n-1] = v[n-1]; for(int i=n-2;i>=0;i--) Max[i] = max(Max[i+1], v[i]); int Min = 1e9; for(int i=0;i<n-1;i++) { Min = min(Min, v[i]); if(Min >= Max[i+1]) { cout << "TAK\n"; cout << i+1 << "\n"; return 0; } } cout << "NIE\n"; return 0; } if(k == 3) { int Max = *max_element(all(v)); int Min = *min_element(all(v)); for(int i=0;i<n-1;i++) { if(v[i] == Max) { cout << "TAK\n"; cout << i << " " << i+1 << "\n"; return 0; } } for(int i=1;i<n;i++) { if(v[i] == Min) { cout << "TAK\n"; cout << i << " " << i+1 << "\n"; return 0; } } cout << "NIE\n"; return 0; } for(int i=0;i<n-1;i++) { if(v[i] >= v[i+1]) { cout << "TAK\n"; int rem = k-4; for(int j=0;j<i-1&&rem;j++) { cout << j+1 << " "; rem--; } cout << i << " " << i+1 << " " << i+2 << " "; for(int j=i+2;j<n&&rem;j++) { cout << j+1 << " "; rem--; } cout << "\n"; return 0; } } cout << "NIE\n"; } |
English