#include<cstdio> #include<vector> #include<map> #include<utility> #include<iostream> #include<algorithm> using namespace std; int main(){ int k, n, t, m, c; int maksymalny = 0; int minimalny = 1000000001; int indeks_minimalnego=0; int indeks_maksymalnego =0; scanf("%d %d", &n, &k); vector<pair<int, int>> v; switch(k){ case 2: for (int i=0; i<n; i++){ scanf("%d", &t); v.push_back({t,i}); } sort(v.begin(), v.end()); m=v[n-1].second; c = 1; for( int i=n-2; i >=0; i--){ if(m == c) break; c++; m = max(m, v[i].second); } if(m == n -1) printf("NIE"); else printf("TAK\n%d", m+1); break; case 3: for (int i=0; i<n; i++){ scanf("%d", &t); if (t > maksymalny){ maksymalny = t; indeks_maksymalnego = i; } if( t == minimalny){ indeks_minimalnego = i; } else if( t < minimalny){ minimalny = t; indeks_minimalnego = i; } } if(indeks_minimalnego == n -1) printf("TAK\n%d %d", indeks_minimalnego-1, indeks_minimalnego); else if (indeks_minimalnego > 0) printf("TAK\n%d %d", indeks_minimalnego, indeks_minimalnego+1); else if (indeks_maksymalnego == 0) printf("TAK\n1 2"); else if (indeks_maksymalnego < n-1) printf("TAK\n%d %d", indeks_maksymalnego, indeks_maksymalnego+1); else printf("NIE"); break; default: int prev = -1; int ideks_separacji = -1; for (int i=0; i<n; i++){ scanf("%d", &t); if( t <= prev){ ideks_separacji = i; break; } prev = t; } if (ideks_separacji == -1){ printf("NIE"); return 0; } printf("TAK\n"); if (ideks_separacji < k) ideks_separacji = 1; if (ideks_separacji >= n - k) ideks_separacji = n -1; if (ideks_separacji == n-1){ for( int i = n-k; i <= n-1; i++) printf("%d ", i); } else{ for( int i = 0; i < k; i++) printf("%d ", ideks_separacji + i); } } }
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 | #include<cstdio> #include<vector> #include<map> #include<utility> #include<iostream> #include<algorithm> using namespace std; int main(){ int k, n, t, m, c; int maksymalny = 0; int minimalny = 1000000001; int indeks_minimalnego=0; int indeks_maksymalnego =0; scanf("%d %d", &n, &k); vector<pair<int, int>> v; switch(k){ case 2: for (int i=0; i<n; i++){ scanf("%d", &t); v.push_back({t,i}); } sort(v.begin(), v.end()); m=v[n-1].second; c = 1; for( int i=n-2; i >=0; i--){ if(m == c) break; c++; m = max(m, v[i].second); } if(m == n -1) printf("NIE"); else printf("TAK\n%d", m+1); break; case 3: for (int i=0; i<n; i++){ scanf("%d", &t); if (t > maksymalny){ maksymalny = t; indeks_maksymalnego = i; } if( t == minimalny){ indeks_minimalnego = i; } else if( t < minimalny){ minimalny = t; indeks_minimalnego = i; } } if(indeks_minimalnego == n -1) printf("TAK\n%d %d", indeks_minimalnego-1, indeks_minimalnego); else if (indeks_minimalnego > 0) printf("TAK\n%d %d", indeks_minimalnego, indeks_minimalnego+1); else if (indeks_maksymalnego == 0) printf("TAK\n1 2"); else if (indeks_maksymalnego < n-1) printf("TAK\n%d %d", indeks_maksymalnego, indeks_maksymalnego+1); else printf("NIE"); break; default: int prev = -1; int ideks_separacji = -1; for (int i=0; i<n; i++){ scanf("%d", &t); if( t <= prev){ ideks_separacji = i; break; } prev = t; } if (ideks_separacji == -1){ printf("NIE"); return 0; } printf("TAK\n"); if (ideks_separacji < k) ideks_separacji = 1; if (ideks_separacji >= n - k) ideks_separacji = n -1; if (ideks_separacji == n-1){ for( int i = n-k; i <= n-1; i++) printf("%d ", i); } else{ for( int i = 0; i < k; i++) printf("%d ", ideks_separacji + i); } } } |