Computer Science/자료구조

[6.5] 프로그래머스 코딩테스트 연습 - K번째 수(2/100)

토마토. 2021. 6. 5. 21:56

코딩테스트 연습 - K번째수 | 프로그래머스 (programmers.co.kr)

 

코딩테스트 연습 - K번째수

[1, 5, 2, 6, 3, 7, 4] [[2, 5, 3], [4, 4, 1], [1, 7, 3]] [5, 6, 3]

programmers.co.kr

def solution(array, commands):
  n = len(array)
  m = len(commands)
  answer = []
  if 1 <= n <= 100 and 1 <= m <= 50:
    for i in range(m):
      if len(commands[i]) == 3:
        lst = array[commands[i][0]-1:commands[i][1]]
        lst.sort()
        if 1<= lst[commands[i][2]-1] <= 100:
          answer.append(lst[commands[i][2]-1])
        else:
          print('error')
      else:
        print('error')

  else:
    print('error')
  return answer