#include<bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<int,int> pii; const int MAXN = 5e5 + 5; const int inf = 0x3f3f3f3f; int a[MAXN]; int main(void) { int n,d; scanf("%d%d",&n,&d); for(int i=1; i<=n; ++i) scanf("%d",&a[i]); if(d >= 4) { for(int i=1; i<n; ++i) if(a[i] >= a[i+1]) { static bool vis[MAXN]; vis[i-1] = vis[i] = vis[i+1] = 1; int cnt = 0; for(int j=1; j<n; ++j) if(vis[j]) ++cnt; for(int j=1; j<n && cnt<d-1; ++j) if(!vis[j]) { vis[j] = 1; ++cnt; } printf("TAK\n"); for(int j=1; j<n; ++j) if(vis[j]) printf("%d ",j); return 0; } printf("NIE\n"); } else if(d == 3) { int mn = *min_element(a+1, a+n+1); int mx = *max_element(a+1, a+n+1); for(int i=2; i<=n; ++i) if(a[i] == mn) { if(i != n) printf("TAK\n%d %d\n",i-1,i); else printf("TAK\n%d %d\n",i-2,i-1); return 0; } for(int i=1; i<=n-1; ++i) if(a[i] == mx) { if(i != 1) printf("TAK\n%d %d\n",i-1,i); else printf("TAK\n%d %d\n",1,2); return 0; } printf("NIE\n"); } else if(d == 2) { static int pre[MAXN], suf[MAXN]; pre[0] = inf; suf[n+1] = -inf; for(int i=1; i<=n; ++i) pre[i] = min(pre[i-1], a[i]); for(int i=n; i>=1; --i) suf[i] = max(suf[i+1], a[i]); for(int i=1; i<n; ++i) if(pre[i] >= suf[i+1]) { printf("TAK\n%d\n",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 | #include<bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<int,int> pii; const int MAXN = 5e5 + 5; const int inf = 0x3f3f3f3f; int a[MAXN]; int main(void) { int n,d; scanf("%d%d",&n,&d); for(int i=1; i<=n; ++i) scanf("%d",&a[i]); if(d >= 4) { for(int i=1; i<n; ++i) if(a[i] >= a[i+1]) { static bool vis[MAXN]; vis[i-1] = vis[i] = vis[i+1] = 1; int cnt = 0; for(int j=1; j<n; ++j) if(vis[j]) ++cnt; for(int j=1; j<n && cnt<d-1; ++j) if(!vis[j]) { vis[j] = 1; ++cnt; } printf("TAK\n"); for(int j=1; j<n; ++j) if(vis[j]) printf("%d ",j); return 0; } printf("NIE\n"); } else if(d == 3) { int mn = *min_element(a+1, a+n+1); int mx = *max_element(a+1, a+n+1); for(int i=2; i<=n; ++i) if(a[i] == mn) { if(i != n) printf("TAK\n%d %d\n",i-1,i); else printf("TAK\n%d %d\n",i-2,i-1); return 0; } for(int i=1; i<=n-1; ++i) if(a[i] == mx) { if(i != 1) printf("TAK\n%d %d\n",i-1,i); else printf("TAK\n%d %d\n",1,2); return 0; } printf("NIE\n"); } else if(d == 2) { static int pre[MAXN], suf[MAXN]; pre[0] = inf; suf[n+1] = -inf; for(int i=1; i<=n; ++i) pre[i] = min(pre[i-1], a[i]); for(int i=n; i>=1; --i) suf[i] = max(suf[i+1], a[i]); for(int i=1; i<n; ++i) if(pre[i] >= suf[i+1]) { printf("TAK\n%d\n",i); return 0; } printf("NIE\n"); } return 0; } |