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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
#include <iostream>
#include <algorithm>
using namespace std;

int main()
{
    int d;
    int n;
    int k;
    cin >> n;
    cin >> k;
    int prz[n];
    for (int i=0;i<n;i++)
    {
        cin >> prz[i];
    }
    int sorted[n];
    for(int i=0;i<n;i++)
    {
        sorted[i] = prz[i];
    }
    sort(sorted,sorted+n);
    if (k>=3  && prz[n-1]!=sorted[n-1])
    {
        d = distance(prz,find(prz,prz+n,sorted[n-1]));
        int z = k-3;
        int g = 1;
        cout << "TAK" << endl;
        while (g<=d-1)
        {
            if(z!=0)
            {
               cout << g << " ";
               z--;
               g++;
            }
            else
            {
                g=d;
            }
        }
        cout << d << " ";
        cout << d+1 << " ";
        g=2;
        while(z!=0)
        {
            cout << d+g << " ";
            z--;
        }

    }
    else if (k==2)
    {
        int j=1;
        int m=n+1;
        for(int i=0;i<n;i++)
        {
            if(prz[i]>=prz[0])
            {
                j=i;
            }
            if(prz[i]<=prz[n-1])
            {
                if(i<m)
                {
                    m=i;
                }
            }
        }
        if(m-j==1)
        {
            cout << "TAK" << endl;
            cout << j+1 ;
        }
        else
        {
            cout << "NIE";
        }
    }
    else
    {
        int a = n-1;
        while(sorted[a]==prz[a])
        {
            a--;
        }
        if(k>=4)
        {
            d = distance(prz,find(prz,prz+n,sorted[a]));
            int z = k-4;
            int g = 1;
            cout << "TAK" << endl;
            while (g<=d-1)
            {
                if(z!=0)
                {
                    cout << g << " ";
                    z--;
                    g++;
                }
                else
                {
                    g=d;
                }
            }
            cout << d << " ";
            cout << d+1 << " ";
            g=2;
            while(z!=0)
            {
                cout << d+g << " ";
                z--;
            }
            cout << a;
        }
        else
        {
            int j=1;
            int m=n+1;
            for(int i=0;i<a;i++)
            {
                if(prz[i]>=prz[0])
                {
                    j=i;
                }
                if(prz[i]<=prz[n-1])
                {
                    if(i<m)
                    {
                        m=i;
                    }
                }
            }
            if(m-j==1)
            {
                cout << "TAK" << endl;
                cout << j+1 ;
                cout << a;
            }
            else
            {
                cout << "NIE";
            }
        }

    }
}