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
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
    ios_base::sync_with_stdio(0);
    int n, k;
    cin >> n >> k;
    vector<int> a(n);
    for (int i = 0; i < n; ++i)
        cin >> a[i];
    if (k == 2)
    {
        vector<int> suf(n);
        suf[n - 1] = a[n - 1];
        for (int i = n - 2; i >= 0; --i)
            suf[i] = max(suf[i + 1], a[i]);
        int pref = a[0];
        for (int i = 0; i < n - 1; ++i)
        {
            pref = min(pref, a[i]);
            if (pref >= suf[i + 1])
            {
                cout << "TAK\n" << i + 1;
                return 0;
            }
        }
        cout << "NIE";
        return 0;
    }
    if (k == 3)
    {
        int i = 1;
        while (i < n && a[i] > a[0])
            ++i;
        if (i < n - 1)
        {
            cout << "TAK\n" << i << " " << i + 1;
            return 0;
        }
        if (i == n - 1)
        {
            cout << "TAK\n" << n - 2 << " " << n - 1;
            return 0;
        }
        i = n - 2;
        while (i >= 0 && a[i] < a[n - 1])
            --i;
        if (i > 0)
        {
            cout << "TAK\n" << i << " " << i + 1;
            return 0;
        }
        if (i == 0)
        {
            cout << "TAK\n1 2";
            return 0;
        }
        cout << "NIE";
        return 0;
    }
    if (k >= 4)
    {
        int zab = 0;
        while (zab < n - 1 && a[zab] < a[zab + 1])
            ++zab;
        if (zab == n - 1)
        {
            cout << "NIE";
            return 0;
        }
        vector<int> odp;
        odp.push_back(zab);
        odp.push_back(zab + 1);
        if (zab > 0 && zab < n - 2)
        {
            odp.push_back(zab - 1);
            odp.push_back(n - 1);
            for (int i = zab - 2; i >= 0 && odp.size() < k; --i)
                odp.push_back(i);
            for (int i = n - 2; i > zab + 1 && odp.size() < k; --i)
                odp.push_back(i);
        }
        if (zab == 0)
        {
            odp.push_back(n - 1);
            for (int i = n - 2; i > zab + 1 && odp.size() < k; --i)
                odp.push_back(i);
        }
        if (zab == n - 2)
        {
            odp.push_back(n - 3);
            for (int i = n - 4; i >= 0 && odp.size() < k; --i)
                odp.push_back(i);
        }
        cout << "TAK\n";
        sort(odp.begin(), odp.end());
        for (int i = 0; i < k - 1; ++i)
            cout << odp[i] + 1 << " ";
        return 0;
    }
}