#include <bits/stdc++.h> using namespace std; const int MAX = 1e9+7; #define pb push_back bool solve(vector<int> &ans, vector<int> &a, int n, int k){ if(k == 2){ vector <int> mn(n), mx(n); mn[0] = a[0]; mx[n-1] = a[n-1]; for(int i = 1; i < n; i++) mn[i] = min(mn[i-1], a[i]); for(int i = n-2; i >= 0; i--) mx[i] = max(mx[i+1], a[i]); for(int i = 0; i < n-1; i++){ if(mn[i] >= mx[i+1]){ ans.pb(i+1); return true; } } return false; } if(k == 3){ int mini = MAX, maxi = 0; for(int i = 0; i < n; i++) { mini = min(mini, a[i]); maxi = max(maxi, a[i]); } if(a[n-1] == mini || a[0] == maxi) { ans.pb(1); ans.pb(n-1); return true; } for(int i = 1; i < n-1; i++){ if(a[i] == mini || a[i] == maxi) { ans.pb(i); ans.pb(i+1); return true; } } return false; } if(k >= 4){ int x = -1; for(int i = 1; i < n; i++){ if(a[i] <= a[i-1]) { x = i+1; break; } } if(x == -1) return false; int y = k-1; if(x != n) { ans.pb(x); y--; } for(int i = x-1; (i > 0 && y > 0); i--) { ans.pb(i); y--; } for(int i = x+1; (i < n && y > 0); i++){ ans.pb(i); y--; } return true; } return false; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k; cin >> n >> k; vector <int> a(n); for(int i = 0; i < n; i++) cin >> a[i]; vector <int> ans; if(solve(ans, a, n, k)){ cout << "TAK\n"; sort(ans.begin(), ans.end()); for(int i = 0; i < ans.size(); i++) cout << ans[i] << " "; cout << "\n"; } else 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 87 88 89 90 | #include <bits/stdc++.h> using namespace std; const int MAX = 1e9+7; #define pb push_back bool solve(vector<int> &ans, vector<int> &a, int n, int k){ if(k == 2){ vector <int> mn(n), mx(n); mn[0] = a[0]; mx[n-1] = a[n-1]; for(int i = 1; i < n; i++) mn[i] = min(mn[i-1], a[i]); for(int i = n-2; i >= 0; i--) mx[i] = max(mx[i+1], a[i]); for(int i = 0; i < n-1; i++){ if(mn[i] >= mx[i+1]){ ans.pb(i+1); return true; } } return false; } if(k == 3){ int mini = MAX, maxi = 0; for(int i = 0; i < n; i++) { mini = min(mini, a[i]); maxi = max(maxi, a[i]); } if(a[n-1] == mini || a[0] == maxi) { ans.pb(1); ans.pb(n-1); return true; } for(int i = 1; i < n-1; i++){ if(a[i] == mini || a[i] == maxi) { ans.pb(i); ans.pb(i+1); return true; } } return false; } if(k >= 4){ int x = -1; for(int i = 1; i < n; i++){ if(a[i] <= a[i-1]) { x = i+1; break; } } if(x == -1) return false; int y = k-1; if(x != n) { ans.pb(x); y--; } for(int i = x-1; (i > 0 && y > 0); i--) { ans.pb(i); y--; } for(int i = x+1; (i < n && y > 0); i++){ ans.pb(i); y--; } return true; } return false; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k; cin >> n >> k; vector <int> a(n); for(int i = 0; i < n; i++) cin >> a[i]; vector <int> ans; if(solve(ans, a, n, k)){ cout << "TAK\n"; sort(ans.begin(), ans.end()); for(int i = 0; i < ans.size(); i++) cout << ans[i] << " "; cout << "\n"; } else cout << "NIE\n"; return 0; } |