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
#include <bits/stdc++.h>
using namespace std;
int n, k;
int mini = 0;
int rosnacy = 0;
int dwojka = false;
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> k;
    int maks = n - 1;
    int tab[n];
    int minimal[n];
    int maksimal[n];
    for (int i = 0; i < n; i++)
        cin >> tab[i];
    minimal[0] = tab[0];
    maksimal[n - 1] = tab[n - 1];
    for (int i = 1; i < n; i++)
    {
        minimal[i] = min(tab[i], minimal[i - 1]);
        if (tab[i] <= tab[mini])
            mini = i;
        if ((tab[i] >= tab[maks]) && (i < n - 1))
            maks = i;
        if (tab[i] <= tab[i - 1])
            rosnacy = i;
    }
    for (int i = n - 2; i >= 0; i--)
    {
        maksimal[i] = max(tab[i], maksimal[i + 1]);
    }
    for (int i = 0; i < n - 1; i++)
    {
        if (minimal[i] >= maksimal[i + 1])
            dwojka = i + 1;
    }
    if (tab[n - 1] == tab[mini])
        mini = n - 1;
    // for (int i = n - 2; i >= 0; i--) // czy k-1 największych jest po kolei?
    // {
    //    // cout << tab[i] << "  " << tab[n-k+1]<< endl;
    //     if (i > n - k)
    //     {
    //        cout << i << "  " << tab[i] <<endl;
    //         if (tab[i] >= tab[i + 1])
    //         {
    //             sukces = false;
    //             break;
    //         }
    //     }
    //     else if (tab[i] >= tab[n - k+1])
    //     {
    //        cout << i << "  " << tab[i] << endl;
    //         sukces = false;
    //         break;
    //     }
    // }
    if (k == 2)
    {
        if (dwojka)
        {
            cout << "TAK" << endl;
            cout << dwojka << endl;
        }
        else
            cout << "NIE" << endl;
    }
    else if (k == 3)
    {
        if ((maks != n - 1) || (mini))
        {
            cout << "TAK" << endl;
            if (maks != n - 1)
                cout << maks << " " << maks + 1 << endl;
            else
                cout << mini << " " << mini + 1 << endl;
        }
        else
            cout << "NIE" << endl;
    }
    else
    {
        if (rosnacy)
        {
            cout << "TAK" << endl;
            if(rosnacy==n-1)
                rosnacy--;
            if (k > rosnacy + 1)
            {
                for (int i = 1; i <= rosnacy; i++)
                    cout << i << " ";
                for (int i = rosnacy+1; i < k; i++)
                    cout << i << " ";
                    cout << endl;
            }
            else
            {
                for (int i = 1; i < k - 3; i++)
                    cout << i << " ";
                cout << rosnacy - 1 << " " << rosnacy << " " << rosnacy+1<< endl;
            }
        }
        else
            cout << "NIE" << endl;
    }
}