#include <bits/stdc++.h> using namespace std; const int MX=500100; int n,m,i,j,a[MX],mnl[MX],mxl[MX],mnr[MX],mxr[MX]; int main() { scanf("%d%d",&n,&m); for (i=1; i<=n; i++) scanf("%d",&a[i]); mnl[1]=mxl[1]=a[1]; for (i=2; i<=n; i++) { mnl[i]=min(mnl[i-1],a[i]); mxl[i]=max(mxl[i-1],a[i]); } mnr[n]=mxr[n]=a[n]; for (i=n-1; i>0; i--) { mnr[i]=min(mnr[i+1],a[i]); mxr[i]=max(mxr[i+1],a[i]); } if (m==2) { for (i=1; i<n; i++) if (mnl[i]>=mxr[i+1]) break; if (i<n) { puts("TAK"); printf("%d\n",i); } else puts("NIE"); return 0; } if (mnl[n]==mxl[n]) { puts("TAK"); for (i=1; i<m; i++) printf("%d ",i); return 0; } if (m>3) { for (i=1; i<n; i++) if (a[i]>=a[i+1]) break; if (i<n) { puts("TAK"); vector<int> ans; if (i>1) ans.push_back(i-1); ans.push_back(i); ans.push_back(i+1); for (j=1; j<i-1 && ans.size()+1<m; j++) ans.push_back(j); for (j=i+2; j<n && ans.size()+1<m; j++) ans.push_back(j); sort(ans.begin(),ans.end()); for (int x: ans) printf("%d ",x); } else puts("NIE"); return 0; } for (j=2; j<n; j++) if (a[j]==mnl[n] || a[j]==mxl[n]) { puts("TAK"); printf("%d %d\n",j-1,j); return 0; } puts("NIE"); 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 | #include <bits/stdc++.h> using namespace std; const int MX=500100; int n,m,i,j,a[MX],mnl[MX],mxl[MX],mnr[MX],mxr[MX]; int main() { scanf("%d%d",&n,&m); for (i=1; i<=n; i++) scanf("%d",&a[i]); mnl[1]=mxl[1]=a[1]; for (i=2; i<=n; i++) { mnl[i]=min(mnl[i-1],a[i]); mxl[i]=max(mxl[i-1],a[i]); } mnr[n]=mxr[n]=a[n]; for (i=n-1; i>0; i--) { mnr[i]=min(mnr[i+1],a[i]); mxr[i]=max(mxr[i+1],a[i]); } if (m==2) { for (i=1; i<n; i++) if (mnl[i]>=mxr[i+1]) break; if (i<n) { puts("TAK"); printf("%d\n",i); } else puts("NIE"); return 0; } if (mnl[n]==mxl[n]) { puts("TAK"); for (i=1; i<m; i++) printf("%d ",i); return 0; } if (m>3) { for (i=1; i<n; i++) if (a[i]>=a[i+1]) break; if (i<n) { puts("TAK"); vector<int> ans; if (i>1) ans.push_back(i-1); ans.push_back(i); ans.push_back(i+1); for (j=1; j<i-1 && ans.size()+1<m; j++) ans.push_back(j); for (j=i+2; j<n && ans.size()+1<m; j++) ans.push_back(j); sort(ans.begin(),ans.end()); for (int x: ans) printf("%d ",x); } else puts("NIE"); return 0; } for (j=2; j<n; j++) if (a[j]==mnl[n] || a[j]==mxl[n]) { puts("TAK"); printf("%d %d\n",j-1,j); return 0; } puts("NIE"); return 0; } |