#include <iostream> #include <stdio.h> #define N 500000 using namespace std; int n,k; int a[N+1],mn[N+1],mx[N+1],mnn=1000000000,mxx; int d=-1; int main() { scanf("%d%d",&n,&k); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); if(a[i]<=a[i-1]) d=i; mnn=min(mnn,a[i]); mxx=max(mxx,a[i]); } if(d==-1) { printf("NIE\n"); return 0; } if(k==2) { mn[1]=a[1]; for(int i=2;i<=n;i++) mn[i]=min(mn[i-1],a[i]); mx[n]=a[n]; for(int i=n-1;i;i--) { if(mn[i]>=mx[i+1]) { //printf("TAK\n"); printf("TAK\n%d\n",i); return 0; } mx[i]=max(mx[i+1],a[i]); } } else if(k==3) { if(a[1]==mxx) { //printf("TAK\n"); printf("TAK\n1 2\n"); return 0; } if(a[n]==mnn) { //printf("TAK\n"); printf("TAK\n%d %d\n",n-2,n-1); return 0; } for(int i=1;i<n-1;i++) { if(a[i+1]==mnn||a[i+1]==mxx) { //printf("TAK\n"); printf("TAK\n%d %d\n",i,i+1); return 0; } } } else { printf("TAK\n"); if(d==n) { for(int i=n-k+1;i<n;i++) printf("%d ",i); } else if(d<k) { for(int i=1;i<k;i++) printf("%d ",i); } else { for(int i=d-k+2;i<=d;i++) printf("%d ",i); } return 0; } printf("NIE\n"); 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 | #include <iostream> #include <stdio.h> #define N 500000 using namespace std; int n,k; int a[N+1],mn[N+1],mx[N+1],mnn=1000000000,mxx; int d=-1; int main() { scanf("%d%d",&n,&k); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); if(a[i]<=a[i-1]) d=i; mnn=min(mnn,a[i]); mxx=max(mxx,a[i]); } if(d==-1) { printf("NIE\n"); return 0; } if(k==2) { mn[1]=a[1]; for(int i=2;i<=n;i++) mn[i]=min(mn[i-1],a[i]); mx[n]=a[n]; for(int i=n-1;i;i--) { if(mn[i]>=mx[i+1]) { //printf("TAK\n"); printf("TAK\n%d\n",i); return 0; } mx[i]=max(mx[i+1],a[i]); } } else if(k==3) { if(a[1]==mxx) { //printf("TAK\n"); printf("TAK\n1 2\n"); return 0; } if(a[n]==mnn) { //printf("TAK\n"); printf("TAK\n%d %d\n",n-2,n-1); return 0; } for(int i=1;i<n-1;i++) { if(a[i+1]==mnn||a[i+1]==mxx) { //printf("TAK\n"); printf("TAK\n%d %d\n",i,i+1); return 0; } } } else { printf("TAK\n"); if(d==n) { for(int i=n-k+1;i<n;i++) printf("%d ",i); } else if(d<k) { for(int i=1;i<k;i++) printf("%d ",i); } else { for(int i=d-k+2;i<=d;i++) printf("%d ",i); } return 0; } printf("NIE\n"); return 0; } |