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
#include <bits/stdc++.h>

using namespace std;

const int N = 500010;

int tab[N], min_prefix[N], max_prefix[N], min_suffix[N], max_suffix[N];

int main() {
	int n, k;
	cin >> n >> k;

	for(int i = 1; i <= n; i++) scanf("%d", tab + i);

	min_suffix[n + 1] = min_prefix[0] = numeric_limits<int>::max();
	max_suffix[n + 1] = max_prefix[0] = 0;

	for(int i = 1; i <= n; i++) {
		max_prefix[i] = max(max_prefix[i -1], tab[i]);
		min_prefix[i] = min(min_prefix[i -1], tab[i]);
	}

	for(int i = n; i > 0; i--) {
		max_suffix[i] = max(max_suffix[i + 1], tab[i]);
		min_suffix[i] = min(min_suffix[i + 1], tab[i]);
	}

	if(k == 2) {
		for(int i = 1; i < n; i++) {
			if(min_prefix[i] >= max_suffix[i + 1]) {
				printf("TAK\n%d\n", i);
				//printf("TAK\n");
				return 0;
			}
		}
		printf("NIE\n");
	} else if(k == 3) {
		for(int i = 2; i < n; i++) {
			if(tab[i] <= min_prefix[i - 1] || tab[i] >= max_suffix[i + 1]) {
				printf("TAK\n%d %d\n", i - 1, i);
				//printf("TAK\n");
				return 0;
			}
		}
		printf("NIE\n");
	} else {
		int x = 0;
		for(int i = 2; i <= n; i++) {
			if(tab[i] <= tab[i - 1]) {
				x = i;
				break;
			}
		}
		if(!x) printf("NIE\n");
		else {
			printf("TAK\n");
			bool tab[N];
			fill(tab, tab + n + 1, false);

			if(x > 2) { tab[x - 2] = true; k--; }
			tab[x - 1] = tab[x] = true;
			k -= 2;

			for(int i = 1; i <= n && k; i++) if(!tab[i]) { tab[i] = true; k--; }

			for(int i = 1; i < n ; i++) if(tab[i]) printf("%d ", i);
			printf("\n");
		}
	}
	return 0;
}