#include <iostream>
#include <vector>
#include <bitset>
using namespace std;
constexpr int N = 5e5 + 7;
bitset <N> ans;
vector <int> ans1;
int tab[N], mx[N], nm[N];
inline void fill_else(int n, int k) {
int i = 0;
while(k) {
if(!ans[i]) {
ans[i] = 1;
k--;
}
i++;
}
}
inline void fill(int x, int n, int k) {
ans[x] = 1; k--;
if(x - 1 >= 0) {
ans[x - 1] = 1;
k--;
} if(x + 1 < n - 1) {
ans[x + 1] = 1;
k--;
}
fill_else(n, k);
return;
}
inline bool case3(int n, int k) {
for(int i = 1; i < n; i++)
if(tab[i - 1] >= tab[i]) {
fill(i - 1, n, k);
return 1;
}
return 0;
}
inline bool case2(int n) {
int k = 2;
for(int i = n - 1; i >= 0; i--)
mx[i] = max(mx[i + 1], tab[i]);
nm[0] = tab[0];
for(int i = 1; i < n; i++)
nm[i] = min(nm[i - 1], tab[i]);
for(int i = 0; i < n - 1; i++)
if(tab[i] >= mx[i + 1]) {
ans[i] = 1; k--;
if(i) {
ans[i - 1] = 1;
k--;
} if(k)
fill_else(n, k);
return 1;
}
for(int i = 1; i < n; i++)
if(tab[i] <= nm[i - 1]) {
ans[i - 1] = 1; k--;
if(i != n - 1) {
ans[i] = 1;
k--;
} if(k)
fill_else(n, k);
}
return 0;
}
inline void case1(int n) {
for(int i = n - 1; i >= 0; i--)
mx[i] = max(mx[i + 1], tab[i]);
nm[0] = tab[0];
for(int i = 1; i < n; i++)
nm[i] = min(nm[i - 1], tab[i]);
for(int i = 0; i < n - 1; i++)
if(nm[i] >= mx[i + 1]) {
ans[i] = 1;
return;
}
}
int main() {
ios_base::sync_with_stdio(0); cin.tie(0);
int n, k; cin >> n >> k;
for(int i = 0; i < n; i++)
cin >> tab[i];
if(k > 3)
case3(n, k - 1);
else if(k == 3)
case2(n);
else
case1(n);
for(int i = 0; i < n - 1; i++)
if(ans[i])
ans1.push_back(i + 1);
if((int)ans1.size() != k - 1) {
cout << "NIE\n";
return 0;
}
cout << "TAK\n";
for(int i = 0; i < k - 1; i++)
cout << ans1[i] << " ";
cout << "\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 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | #include <iostream> #include <vector> #include <bitset> using namespace std; constexpr int N = 5e5 + 7; bitset <N> ans; vector <int> ans1; int tab[N], mx[N], nm[N]; inline void fill_else(int n, int k) { int i = 0; while(k) { if(!ans[i]) { ans[i] = 1; k--; } i++; } } inline void fill(int x, int n, int k) { ans[x] = 1; k--; if(x - 1 >= 0) { ans[x - 1] = 1; k--; } if(x + 1 < n - 1) { ans[x + 1] = 1; k--; } fill_else(n, k); return; } inline bool case3(int n, int k) { for(int i = 1; i < n; i++) if(tab[i - 1] >= tab[i]) { fill(i - 1, n, k); return 1; } return 0; } inline bool case2(int n) { int k = 2; for(int i = n - 1; i >= 0; i--) mx[i] = max(mx[i + 1], tab[i]); nm[0] = tab[0]; for(int i = 1; i < n; i++) nm[i] = min(nm[i - 1], tab[i]); for(int i = 0; i < n - 1; i++) if(tab[i] >= mx[i + 1]) { ans[i] = 1; k--; if(i) { ans[i - 1] = 1; k--; } if(k) fill_else(n, k); return 1; } for(int i = 1; i < n; i++) if(tab[i] <= nm[i - 1]) { ans[i - 1] = 1; k--; if(i != n - 1) { ans[i] = 1; k--; } if(k) fill_else(n, k); } return 0; } inline void case1(int n) { for(int i = n - 1; i >= 0; i--) mx[i] = max(mx[i + 1], tab[i]); nm[0] = tab[0]; for(int i = 1; i < n; i++) nm[i] = min(nm[i - 1], tab[i]); for(int i = 0; i < n - 1; i++) if(nm[i] >= mx[i + 1]) { ans[i] = 1; return; } } int main() { ios_base::sync_with_stdio(0); cin.tie(0); int n, k; cin >> n >> k; for(int i = 0; i < n; i++) cin >> tab[i]; if(k > 3) case3(n, k - 1); else if(k == 3) case2(n); else case1(n); for(int i = 0; i < n - 1; i++) if(ans[i]) ans1.push_back(i + 1); if((int)ans1.size() != k - 1) { cout << "NIE\n"; return 0; } cout << "TAK\n"; for(int i = 0; i < k - 1; i++) cout << ans1[i] << " "; cout << "\n"; } |
English