#include <cstdio>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <list>
#include <vector>
#include <map>
#include <set>
#include <string>
#include <cstring>
#include <cassert>
#include <limits.h>
#include <iterator>
using namespace std;
typedef long long LL;
typedef long double LD;
typedef vector<int> VI;
typedef pair<int,int> PII;
#define REP(i,n) for(int i=0;i<(n);i++)
#define FOR(i,a,b) for(VAR(i,a);i<=(b);++i)
#define FORD(i,a,b) for(VAR(i,a);i>=(b);--i)
#define FORE(a,b) for(VAR(a,(b).begin());a!=(b).end();++a)
#define VAR(a,b) __typeof(b) a=(b)
#define SIZE(a) ((int)((a).size()))
#define ALL(x) (x).begin(),(x).end()
#define CLR(x,a) memset(x,a,sizeof(x))
#define ZERO(x) memset(x,0,sizeof(x))
#define S(t,i) scanf("%" ## t, &i)
#define SI(i) scanf("%d", &i)
const int INF=INT_MAX;
const int MAXN=1000000+5;
int n, k;
int a[MAXN];
int main() {
ios_base::sync_with_stdio(0);
SI(n);
SI(k);
FOR(i, 1, n) {
SI(a[i]);
}
if (k == 2) {
int minPref = a[1];
multiset<int> suffix(&a[2], &a[n+1]);
// suffix.erase(suffix.find(2));
// FORE(it, suffix) {
// printf("%d ", *it);
// }
// printf("\n%d ", *suffix.rbegin());
// return 0;
FOR(i, 2, n) {
if (minPref >= *suffix.rbegin()) {
printf("TAK\n%d\n", i-1);
return 0;
}
minPref = min(minPref, a[i]);
suffix.erase(suffix.find(a[i]));
}
printf("NIE\n");
return 0;
} else if (k == 3) {
int minA = a[1];
FOR(i, 2, n-1) {
if (a[i] <= minA) {
printf("TAK\n%d %d\n", i-1, i);
return 0;
}
}
if (a[n] <= minA) {
printf("TAK\n%d %d\n", n-2, n-1);
return 0;
}
int maxA = a[n];
FORD(i, n-1, 2) {
if (a[i] >= maxA) {
printf("TAK\n%d %d\n", i-1, i);
return 0;
}
}
if (a[1] >= maxA) {
printf("TAK\n%d %d\n", 1, 2);
return 0;
}
printf("NIE\n");
return 0;
} else {
int prev = a[1];
FOR(i, 2, n) {
if (prev >= a[i]) {
printf("TAK\n");
int poczatek = max(1, i - (k-1) + 1);
FOR(wi, poczatek, max(i, k-1)) {
printf("%d ", wi);
}
return 0;
}
prev = a[i];
}
printf("NIE\n");
return 0;
}
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 | #include <cstdio> #include <cmath> #include <iostream> #include <algorithm> #include <queue> #include <list> #include <vector> #include <map> #include <set> #include <string> #include <cstring> #include <cassert> #include <limits.h> #include <iterator> using namespace std; typedef long long LL; typedef long double LD; typedef vector<int> VI; typedef pair<int,int> PII; #define REP(i,n) for(int i=0;i<(n);i++) #define FOR(i,a,b) for(VAR(i,a);i<=(b);++i) #define FORD(i,a,b) for(VAR(i,a);i>=(b);--i) #define FORE(a,b) for(VAR(a,(b).begin());a!=(b).end();++a) #define VAR(a,b) __typeof(b) a=(b) #define SIZE(a) ((int)((a).size())) #define ALL(x) (x).begin(),(x).end() #define CLR(x,a) memset(x,a,sizeof(x)) #define ZERO(x) memset(x,0,sizeof(x)) #define S(t,i) scanf("%" ## t, &i) #define SI(i) scanf("%d", &i) const int INF=INT_MAX; const int MAXN=1000000+5; int n, k; int a[MAXN]; int main() { ios_base::sync_with_stdio(0); SI(n); SI(k); FOR(i, 1, n) { SI(a[i]); } if (k == 2) { int minPref = a[1]; multiset<int> suffix(&a[2], &a[n+1]); // suffix.erase(suffix.find(2)); // FORE(it, suffix) { // printf("%d ", *it); // } // printf("\n%d ", *suffix.rbegin()); // return 0; FOR(i, 2, n) { if (minPref >= *suffix.rbegin()) { printf("TAK\n%d\n", i-1); return 0; } minPref = min(minPref, a[i]); suffix.erase(suffix.find(a[i])); } printf("NIE\n"); return 0; } else if (k == 3) { int minA = a[1]; FOR(i, 2, n-1) { if (a[i] <= minA) { printf("TAK\n%d %d\n", i-1, i); return 0; } } if (a[n] <= minA) { printf("TAK\n%d %d\n", n-2, n-1); return 0; } int maxA = a[n]; FORD(i, n-1, 2) { if (a[i] >= maxA) { printf("TAK\n%d %d\n", i-1, i); return 0; } } if (a[1] >= maxA) { printf("TAK\n%d %d\n", 1, 2); return 0; } printf("NIE\n"); return 0; } else { int prev = a[1]; FOR(i, 2, n) { if (prev >= a[i]) { printf("TAK\n"); int poczatek = max(1, i - (k-1) + 1); FOR(wi, poczatek, max(i, k-1)) { printf("%d ", wi); } return 0; } prev = a[i]; } printf("NIE\n"); return 0; } return 0; } |
English