K번째수 - Python

1. 풀이 코드

def solution(array, commands):
    answer = []
    for item in commands:
        temp = array[item[0] - 1:item[1]]
        temp.sort()
        answer.append(temp[item[2] - 1])
    return answer


# Test Cases
print(solution([1, 5, 2, 6, 3, 7, 4], [[2, 5, 3], [4, 4, 1], [1, 7, 3]]))

  차근차근 풀었습니다.

 

2. 다른 사람 풀이 코드

def solution(array, commands):
    return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))

  한 줄로 표현하다니 대단하네요.

 

---

• 프로그래머스: https://school.programmers.co.kr/learn/courses/30/lessons/42748

프로그래머스