// 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; } |