// Jakub Rozek
// Podwyzki
// PA 2022 B r2
// czas: n
// pami: n
#include "bits/stdc++.h"
using namespace std;
const int N=500005;
int n,k;
int t[N];
void solvk4()
{
for(int i=2; i<=n; ++i)
{
if(t[i-1]>=t[i])
{
t[i-1]=-1;
--k;
if(i>2)
{
t[i-2]=-1;
--k;
}
if(i<n)
{
t[i]=-1;
--k;
}
}
}
for(int i=1; i<=n; ++i)
{
if(k==1) break;
if(t[i]>0)
{
t[i]=-1;
--k;
}
}
cout<<"TAK\n";
for(int i=1; i<=n; ++i)
{
if(t[i]<0) cout<<i<<" ";
}
}
void solvk3()
{
for(int i=2; i<n; ++i)
{
if(t[1]>=t[i])
{
cout<<"TAK\n";
cout<<i-1<<" "<<i;
return;
}
}
for(int i=n-1; i>=1; --i)
{
if(t[i]>=t[n])
{
cout<<"TAK\n";
cout<<i-1<<" "<<i;
return;
}
}
cout<<"NIE\n";
}
void solvk2()
{
int tab[n+1][2];
tab[1][0]=t[1];
tab[n][1]=t[n];
for(int i=2; i<=n; ++i) tab[i][0]=min(tab[i-1][0],t[i]);
for(int i=n-1; i>=1; --i) tab[i][1]=max(tab[i+1][1],t[i]);
for(int i=1; i<n; ++i)
{
if(tab[i][0]>=tab[i+1][1])
{
cout<<"TAK\n";
cout<<i;
return;
}
}
cout<<"NIE\n";
}
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cin>>n>>k;
for(int i=1; i<=n; ++i) cin>>t[i];
bool f=1;
for(int i=2; i<=n; ++i)
{
if(t[i-1]>=t[i]) f=0;
}
if(f)
{
cout<<"NIE\n";
return 0;
}
if(k==2) solvk2();
else if(k==3) solvk3();
else solvk4();
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 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 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 | // Jakub Rozek // Podwyzki // PA 2022 B r2 // czas: n // pami: n #include "bits/stdc++.h" using namespace std; const int N=500005; int n,k; int t[N]; void solvk4() { for(int i=2; i<=n; ++i) { if(t[i-1]>=t[i]) { t[i-1]=-1; --k; if(i>2) { t[i-2]=-1; --k; } if(i<n) { t[i]=-1; --k; } } } for(int i=1; i<=n; ++i) { if(k==1) break; if(t[i]>0) { t[i]=-1; --k; } } cout<<"TAK\n"; for(int i=1; i<=n; ++i) { if(t[i]<0) cout<<i<<" "; } } void solvk3() { for(int i=2; i<n; ++i) { if(t[1]>=t[i]) { cout<<"TAK\n"; cout<<i-1<<" "<<i; return; } } for(int i=n-1; i>=1; --i) { if(t[i]>=t[n]) { cout<<"TAK\n"; cout<<i-1<<" "<<i; return; } } cout<<"NIE\n"; } void solvk2() { int tab[n+1][2]; tab[1][0]=t[1]; tab[n][1]=t[n]; for(int i=2; i<=n; ++i) tab[i][0]=min(tab[i-1][0],t[i]); for(int i=n-1; i>=1; --i) tab[i][1]=max(tab[i+1][1],t[i]); for(int i=1; i<n; ++i) { if(tab[i][0]>=tab[i+1][1]) { cout<<"TAK\n"; cout<<i; return; } } cout<<"NIE\n"; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin>>n>>k; for(int i=1; i<=n; ++i) cin>>t[i]; bool f=1; for(int i=2; i<=n; ++i) { if(t[i-1]>=t[i]) f=0; } if(f) { cout<<"NIE\n"; return 0; } if(k==2) solvk2(); else if(k==3) solvk3(); else solvk4(); return 0; } |
English