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
//Mateusz Kruk
#include <bits/stdc++.h>
using namespace std;
int main(){
    int n,k;
    cin>>n>>k;
    vector<int>A;
    for(int i=0;i<n;i++){
        int a;
        cin>>a;
        A.push_back(a);
    }
    int j=1;
    while(A[j]>A[0]&&n-j>=k){
        j++;
    }
    vector<int>ans;
    if(A[j]<=A[0]){
        if(k!=2){
            cout<<"TAK"<<'\n';
            cout<<j<<" ";
            int x=k-2;
            for(int h=j+1;h<j+1+x;h++){
                cout<<h<<" ";
            }
            cout<<'\n';
        }
        else{
            bool e=true;
            int h=j+1;
            while(j<n){
                if(A[j]>A[0]){
                    e=false;
                }
                j++;
            }
            if(e==true){
                cout<<"TAK"<<'\n';
                cout<<h-1<<'\n';
            }
            else{
                cout<<"NIE"<<'\n';
            }
        }
    }
    else{
        ans.push_back(j);
        bool e=false;
        for(int i=j+1;i<n;i++){
            ans.push_back(i);
            if(A[i-1]>A[i]){
                e=true;
            }
        }
        if(e==true){
            cout<<"TAK"<<'\n';
            for(int i=0;i<ans.size();i++){
                cout<<ans[i]<<" ";
            }
            cout<<'\n';
        }
        else{
            cout<<"NIE"<<'\n';
        }
    }
}