#include<cstdio>
#include<vector>
void for_k23(std::vector<int> &tab, int k) {
std::vector<int> prefix_min(tab.size(), 0);
prefix_min[0] = tab[0];
for (int i = 1; i < tab.size(); ++i)
prefix_min[i] = std::min(prefix_min[i-1], tab[i]);
std::vector<int> suffix_max(tab.size(), 0);
suffix_max[tab.size()-1] = tab[tab.size()-1];
for (int i = tab.size() - 2; i >= 0; --i)
suffix_max[i] = std::max(suffix_max[i+1], tab[i]);
if (k == 2) {
for (int i = 1; i < tab.size(); ++i) {
//fprintf(stderr, "idx:%d pmin:%d smax:%d\n", i, prefix_min[i], suffix_max[i]);
if (prefix_min[i-1] >= suffix_max[i]) {
printf("TAK\n%d\n", i);
return;
}
}
printf("NIE\n");
} else { // k == 3
for (int i = 1; i < tab.size()-1; ++i) {
//printf("idx:%d pmin:%d smax:%d\n", i, prefix_min[i], suffix_max[i]);
if (prefix_min[i] >= tab[i]) {
printf("TAK\n%d %d\n", i, i+1); // koniec to i - 1, + 1 (index od 1)
return;
}
}
for (int i = 1; i < tab.size()-1; ++i) {
//printf("idx:%d pmin:%d smax:%d\n", i, prefix_min[i], suffix_max[i]);
if (suffix_max[i] <= tab[i]) {
printf("TAK\n%d %d\n", i, i+1); // koniec to i - 1, + 1 (index od 1)
return;
}
}
printf("NIE\n");
}
}
void for_big_k(std::vector<int> &tab, int k) {
for (int i = 1; i < tab.size(); ++i) {
if (tab[i-1] >= tab[i]) {
printf("TAK\n");
for (int j = 1; j < i-1 && j < k-2; ++j)
printf("%d ", j);
printf("%d ", i);
printf("%d ", i+1);
for (int j = i+2; j < k; ++j)
printf("%d ", j);
printf("\n");
return;
}
}
printf("NIE\n");
}
int main() {
int n, k;
scanf("%d %d", &n, &k);
std::vector<int> tab(n);
for (int i = 0; i < n; ++i)
scanf("%d", &tab[i]);
if (k <= 3) for_k23(tab, k);
else for_big_k(tab, k);
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 | #include<cstdio> #include<vector> void for_k23(std::vector<int> &tab, int k) { std::vector<int> prefix_min(tab.size(), 0); prefix_min[0] = tab[0]; for (int i = 1; i < tab.size(); ++i) prefix_min[i] = std::min(prefix_min[i-1], tab[i]); std::vector<int> suffix_max(tab.size(), 0); suffix_max[tab.size()-1] = tab[tab.size()-1]; for (int i = tab.size() - 2; i >= 0; --i) suffix_max[i] = std::max(suffix_max[i+1], tab[i]); if (k == 2) { for (int i = 1; i < tab.size(); ++i) { //fprintf(stderr, "idx:%d pmin:%d smax:%d\n", i, prefix_min[i], suffix_max[i]); if (prefix_min[i-1] >= suffix_max[i]) { printf("TAK\n%d\n", i); return; } } printf("NIE\n"); } else { // k == 3 for (int i = 1; i < tab.size()-1; ++i) { //printf("idx:%d pmin:%d smax:%d\n", i, prefix_min[i], suffix_max[i]); if (prefix_min[i] >= tab[i]) { printf("TAK\n%d %d\n", i, i+1); // koniec to i - 1, + 1 (index od 1) return; } } for (int i = 1; i < tab.size()-1; ++i) { //printf("idx:%d pmin:%d smax:%d\n", i, prefix_min[i], suffix_max[i]); if (suffix_max[i] <= tab[i]) { printf("TAK\n%d %d\n", i, i+1); // koniec to i - 1, + 1 (index od 1) return; } } printf("NIE\n"); } } void for_big_k(std::vector<int> &tab, int k) { for (int i = 1; i < tab.size(); ++i) { if (tab[i-1] >= tab[i]) { printf("TAK\n"); for (int j = 1; j < i-1 && j < k-2; ++j) printf("%d ", j); printf("%d ", i); printf("%d ", i+1); for (int j = i+2; j < k; ++j) printf("%d ", j); printf("\n"); return; } } printf("NIE\n"); } int main() { int n, k; scanf("%d %d", &n, &k); std::vector<int> tab(n); for (int i = 0; i < n; ++i) scanf("%d", &tab[i]); if (k <= 3) for_k23(tab, k); else for_big_k(tab, k); return 0; } |
English