#include <iostream>
#include <vector>
using namespace std;
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int n;
cin >> n;
int k;
cin >> k;
bool ok = 0;
int last = 0;
vector <int> tab;
for (int i = 0; i < n; i++)
{
int x;
cin >> x;
tab.push_back(x);
}
int act = tab[0];
int indeks = 0;
for (int i = 1; i < n; i++)
{
int x = tab[i];
if (x <= act)
{
//cout << n - i << " " << k - 1;
if (k == 2)
{
int ok2 = 0;
for (int j = i; j < n; j++)
{
if (tab[j] > act)
{
ok2 = 1;
}
}
if(ok2 == 0)
{
indeks = i;
ok = 1;
cout << "TAK";
cout << "\n" << indeks + 1;
return 0;
}
}
if (n - i <= k - 1)
{
ok = 1;
cout << "TAK" << "\n";
for(int j = 2; j <= n; j++)
{
k--;
if(k == 2)
{
cout << i << " " << i + 1 << " ";
return 0;
}
if(j == i)
{
cout << i << " ";
k--;
}
if(j == i + 1)
{
cout << i + 1 << " ";
k--;
}
else if(j!= i + 1 && j!= i)
{
cout << j << " ";
k--;
}
}
break;
}
act = x;
}
}
if (ok == 0)
{
cout << "NIE";
}
}
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 | #include <iostream> #include <vector> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n; cin >> n; int k; cin >> k; bool ok = 0; int last = 0; vector <int> tab; for (int i = 0; i < n; i++) { int x; cin >> x; tab.push_back(x); } int act = tab[0]; int indeks = 0; for (int i = 1; i < n; i++) { int x = tab[i]; if (x <= act) { //cout << n - i << " " << k - 1; if (k == 2) { int ok2 = 0; for (int j = i; j < n; j++) { if (tab[j] > act) { ok2 = 1; } } if(ok2 == 0) { indeks = i; ok = 1; cout << "TAK"; cout << "\n" << indeks + 1; return 0; } } if (n - i <= k - 1) { ok = 1; cout << "TAK" << "\n"; for(int j = 2; j <= n; j++) { k--; if(k == 2) { cout << i << " " << i + 1 << " "; return 0; } if(j == i) { cout << i << " "; k--; } if(j == i + 1) { cout << i + 1 << " "; k--; } else if(j!= i + 1 && j!= i) { cout << j << " "; k--; } } break; } act = x; } } if (ok == 0) { cout << "NIE"; } } |
English