n, k = map(int, input().split()) l = list(map(int, input().split())) kk = k if k >= 4: b = [0] * (n + 10) for i in range(n - 1): if l[i + 1] <= l[i]: b[i] = 1 b[i + 1] = 1 k -= 2 if i: b[i - 1] = 1 k -= 1 break else: print('NIE') exit() o = [] for i in range(n): if k > 1 and b[i] == 0: k -= 1 b[i] = 1 if b[i]: o.append(i + 1) assert len(o) == kk - 1 print('TAK') print(' '.join(map(str, o))) elif k == 3: for i in range(1, n - 1): if l[i] <= l[0]: print('TAK') print(i, i + 1) exit() for i in range(n - 2, 0, -1): if l[i] >= l[-1]: print('TAK') print(i, i + 1) exit() print('NIE') else: assert k == 2 mins = [l[0]] for i in range(1, n - 1): mins.append(min(mins[i - 1], l[i])) maxs = [l[-1]] for i in range(n - 1, -1, -1): maxs.append(max(maxs[-1], l[i])) maxs.reverse() for i in range(n - 1): if mins[i] >= maxs[i + 1]: print('TAK') print(i + 1) exit() print('NIE')
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 | n, k = map(int, input().split()) l = list(map(int, input().split())) kk = k if k >= 4: b = [0] * (n + 10) for i in range(n - 1): if l[i + 1] <= l[i]: b[i] = 1 b[i + 1] = 1 k -= 2 if i: b[i - 1] = 1 k -= 1 break else: print('NIE') exit() o = [] for i in range(n): if k > 1 and b[i] == 0: k -= 1 b[i] = 1 if b[i]: o.append(i + 1) assert len(o) == kk - 1 print('TAK') print(' '.join(map(str, o))) elif k == 3: for i in range(1, n - 1): if l[i] <= l[0]: print('TAK') print(i, i + 1) exit() for i in range(n - 2, 0, -1): if l[i] >= l[-1]: print('TAK') print(i, i + 1) exit() print('NIE') else: assert k == 2 mins = [l[0]] for i in range(1, n - 1): mins.append(min(mins[i - 1], l[i])) maxs = [l[-1]] for i in range(n - 1, -1, -1): maxs.append(max(maxs[-1], l[i])) maxs.reverse() for i in range(n - 1): if mins[i] >= maxs[i + 1]: print('TAK') print(i + 1) exit() print('NIE') |