#include<iostream> using namespace std; int n,k; int t[600000]; int minn[600000]; int maxx[600000]; bool biore[600000]; void licz(int x) { minn[1]=t[1]; for(int i=2;i<=x;i++) minn[i]=min(minn[i-1],t[i]); maxx[x+1]=0; for(int i=x;i>=1;i--) maxx[i]=max(maxx[i+1],t[i]); } void rozw2() { licz(n); for(int i=1;i<n;i++) { if(minn[i]>=maxx[i+1]) { cout<<"TAK\n"<<i; return; } } cout<<"NIE"; } void rozw3() { licz(n); int max=maxx[1]; int gdzie=0; for(int i=2;i<n;i++) { if(t[i]==max) { gdzie=i; break; } } if(gdzie!=0) { cout<<"TAK\n"<<gdzie-1<<" "<<gdzie; return; } if(t[1]==max) { cout<<"TAK\n1 2\n"; return; } else{ licz(n-1); for(int i=1;i<n-1;i++) { if(minn[i]>=maxx[i+1]) { cout<<"TAK\n"<<i<<" "<<n-1; return; } } } cout<<"NIE"; } void rozw4() { bool git=0; int suma=0; for(int i=1;i<n;i++) { if(t[i]>=t[i+1]) { if(i>1) biore[i-1]=1; biore[i]=1; if(i<n-1) biore[i+1]=1; git=1; suma=biore[i-1]+biore[i]+biore[i+1]; break; } } if(git==0) { cout<<"NIE"; return; } cout<<"TAK\n"; for(int i=1;i<n;i++) { if(biore[i]) cout<<i<<" "; else if(suma<k-1) { suma++; cout<<i<<" "; } } } int main() { cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(0); cin>>n>>k; for(int i=1;i<=n;i++) cin>>t[i]; if(k==2) rozw2(); else if(k==3) rozw3(); else rozw4(); 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 | #include<iostream> using namespace std; int n,k; int t[600000]; int minn[600000]; int maxx[600000]; bool biore[600000]; void licz(int x) { minn[1]=t[1]; for(int i=2;i<=x;i++) minn[i]=min(minn[i-1],t[i]); maxx[x+1]=0; for(int i=x;i>=1;i--) maxx[i]=max(maxx[i+1],t[i]); } void rozw2() { licz(n); for(int i=1;i<n;i++) { if(minn[i]>=maxx[i+1]) { cout<<"TAK\n"<<i; return; } } cout<<"NIE"; } void rozw3() { licz(n); int max=maxx[1]; int gdzie=0; for(int i=2;i<n;i++) { if(t[i]==max) { gdzie=i; break; } } if(gdzie!=0) { cout<<"TAK\n"<<gdzie-1<<" "<<gdzie; return; } if(t[1]==max) { cout<<"TAK\n1 2\n"; return; } else{ licz(n-1); for(int i=1;i<n-1;i++) { if(minn[i]>=maxx[i+1]) { cout<<"TAK\n"<<i<<" "<<n-1; return; } } } cout<<"NIE"; } void rozw4() { bool git=0; int suma=0; for(int i=1;i<n;i++) { if(t[i]>=t[i+1]) { if(i>1) biore[i-1]=1; biore[i]=1; if(i<n-1) biore[i+1]=1; git=1; suma=biore[i-1]+biore[i]+biore[i+1]; break; } } if(git==0) { cout<<"NIE"; return; } cout<<"TAK\n"; for(int i=1;i<n;i++) { if(biore[i]) cout<<i<<" "; else if(suma<k-1) { suma++; cout<<i<<" "; } } } int main() { cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(0); cin>>n>>k; for(int i=1;i<=n;i++) cin>>t[i]; if(k==2) rozw2(); else if(k==3) rozw3(); else rozw4(); return 0; } |