#include<iostream> #include<map> #include<set> using namespace std; int T[1000007]; int f(int n, int p, int a){ //a bedie 0 int maks=0, minn=0; for (int i=a; i<n-a; i++){ if (T[i]>=T[maks]) maks=i; if (T[i]<T[minn]) minn=i; } //cout<<maks<<" "<<minn<<"\n"; if (maks!=n-1-a){ cout<<"TAK \n"; for (int i=maks-p+3; i<maks+2; i++){ cout<<i<<" "; } cout<<"\n"; return 0; } if (minn!=a){ cout<<"TAK \n"; for (int i=minn; i<minn+p-1; i++){ cout<<i<<" "; } cout<<"\n"; return 0; } else if (p-2>1){ f(n, p-2, a+1); } else{ cout<<"NIE \n"; return 0; } } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); int n, p; cin>>n>>p; for (int i=0; i<n; i++) cin>>T[i]; f(n, p, 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 | #include<iostream> #include<map> #include<set> using namespace std; int T[1000007]; int f(int n, int p, int a){ //a bedie 0 int maks=0, minn=0; for (int i=a; i<n-a; i++){ if (T[i]>=T[maks]) maks=i; if (T[i]<T[minn]) minn=i; } //cout<<maks<<" "<<minn<<"\n"; if (maks!=n-1-a){ cout<<"TAK \n"; for (int i=maks-p+3; i<maks+2; i++){ cout<<i<<" "; } cout<<"\n"; return 0; } if (minn!=a){ cout<<"TAK \n"; for (int i=minn; i<minn+p-1; i++){ cout<<i<<" "; } cout<<"\n"; return 0; } else if (p-2>1){ f(n, p-2, a+1); } else{ cout<<"NIE \n"; return 0; } } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); int n, p; cin>>n>>p; for (int i=0; i<n; i++) cin>>T[i]; f(n, p, 0); } |