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

int tab[500010];
int mini[500010];
int maxi[500010];
bool zaznaczone[500010];
int main()
{
	int n, k, i;
	scanf("%d%d", &n, &k);
	//printf("%d %d\n", n, k);
	mini[0] = 1000000010;
	for(i=1;i<=n;i++){
		scanf("%d", &tab[i]);
		//printf("%d ", tab[i]);
		mini[i] = min(mini[i-1], tab[i]);
	}
	//printf("\n");
	for(i=n;i>=1;i--){
		maxi[i] = max(tab[i], maxi[i+1]);
	}
	if(k==2){
		for(i=1;i<=n-1;i++){
			if(mini[i]>=maxi[i+1]){
				printf("TAK\n%d\n", i);
				return 0;
			}
		}
		printf("NIE\n");
		return 0;
	}
	if(k==3){
		for(i=2;i<=n-1;i++){
			if(mini[i-1]>=tab[i] || tab[i]>=maxi[i+1]){
				printf("TAK\n%d %d\n", i-1, i);
				return 0;
			}
		}
		printf("NIE\n");
		return 0;
	}
	//k>=4
	int ilosc = 0;
	for(i=1;i<n;i++){
		if(tab[i]>=tab[i+1]){
			zaznaczone[i]=1;
			if(i<n-1){
				ilosc++;
				zaznaczone[i+1]=1;
			}
			if(i>1){
				zaznaczone[i-1]=1;
				ilosc++;
			}
			ilosc+=1;
			break;
		}
	}
	if(ilosc==0){
		printf("NIE\n");
		return 0;
	}
	vector<int>odp;
	for(i=1;i<n;i++){
		//printf("%d\n", ilosc);
		if(zaznaczone[i]==0 && ilosc+1!=k){
			ilosc++;
			zaznaczone[i] = 1;
		}
		if(zaznaczone[i])
			odp.push_back(i);
	}
	printf("TAK\n");
	for(auto j: odp)
		printf("%d ", j);
	return 0;
}

//dla k>=4 trzeba sprawdzic czy jest scisle monotoniczny(para  nieposortowana)
//dla k=2 mozna brutalnie
//dla k=3 analogicznie do 2, srodkowy jest dlugosci 1