#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); } } } |
English