#include<bits/stdc++.h> using namespace std; int n,k,a[500001],najw[500002]; bool ck[500001]; int main() { scanf("%d%d",&n,&k); a[0]=INT_MIN; int gdzie=0; for(int i=1;i<=n;++i) { scanf("%d",&a[i]); if(a[i]<=a[i-1]) gdzie=i; } if(gdzie==0) { printf("NIE"); return 0; } if(k==2) { najw[n+1]=INT_MIN; for(int i=n;i>1;--i) najw[i]=max(najw[i+1],a[i]); int najm=INT_MAX; for(int i=1;i<n;++i) { najm=min(najm,a[i]); if(najm>=najw[i+1]) { printf("TAK\n%d",i); return 0; } } printf("NIE"); } else if(k==3) { for(int i=n;i>1;--i) najw[i]=max(najw[i+1],a[i]); int najm=a[1]; for(int i=2;i<n;++i) { if(a[i]>=najw[i+1]||najm>=a[i]) { printf("TAK\n%d %d",i-1,i); return 0; } najm=min(najm,a[i]); } printf("NIE"); } else { if(gdzie==2) { ck[1]=1; ck[2]=1; k-=2; for(int i=3;k>1;++i) { --k; ck[i]=1; } } else if(gdzie==n) { ck[n-1]=1; k-=2; for(int i=n-2;k;--i) { --k; ck[i]=1; } } else { k-=2; int lew=gdzie-2,praw=gdzie+1; ck[gdzie]=1; ck[gdzie-1]=1; while(lew>1&&k>=2) { --k; ck[lew]=1; --lew; } while(praw<n&&k>=2) { --k; ck[praw]=1; ++praw; } ck[lew]=1; } printf("TAK\n"); for(int i=1;i<=n;++i) if(ck[i]) printf("%d ",i); } 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 | #include<bits/stdc++.h> using namespace std; int n,k,a[500001],najw[500002]; bool ck[500001]; int main() { scanf("%d%d",&n,&k); a[0]=INT_MIN; int gdzie=0; for(int i=1;i<=n;++i) { scanf("%d",&a[i]); if(a[i]<=a[i-1]) gdzie=i; } if(gdzie==0) { printf("NIE"); return 0; } if(k==2) { najw[n+1]=INT_MIN; for(int i=n;i>1;--i) najw[i]=max(najw[i+1],a[i]); int najm=INT_MAX; for(int i=1;i<n;++i) { najm=min(najm,a[i]); if(najm>=najw[i+1]) { printf("TAK\n%d",i); return 0; } } printf("NIE"); } else if(k==3) { for(int i=n;i>1;--i) najw[i]=max(najw[i+1],a[i]); int najm=a[1]; for(int i=2;i<n;++i) { if(a[i]>=najw[i+1]||najm>=a[i]) { printf("TAK\n%d %d",i-1,i); return 0; } najm=min(najm,a[i]); } printf("NIE"); } else { if(gdzie==2) { ck[1]=1; ck[2]=1; k-=2; for(int i=3;k>1;++i) { --k; ck[i]=1; } } else if(gdzie==n) { ck[n-1]=1; k-=2; for(int i=n-2;k;--i) { --k; ck[i]=1; } } else { k-=2; int lew=gdzie-2,praw=gdzie+1; ck[gdzie]=1; ck[gdzie-1]=1; while(lew>1&&k>=2) { --k; ck[lew]=1; --lew; } while(praw<n&&k>=2) { --k; ck[praw]=1; ++praw; } ck[lew]=1; } printf("TAK\n"); for(int i=1;i<=n;++i) if(ck[i]) printf("%d ",i); } return 0; } |