배열의 원하는 index를 잘라서 가져오는 정도의 간단한 문제이다.
Lambda 함수
3.5 람다(lambda) - 왕초보를 위한 Python: 쉽게 풀어 쓴 기초 문법과 실습 (wikidocs.net)
코드
def solution(array, commands):
answer = []
for command in commands:
answer.append(sorted(array[(command[0]-1):command[1]])[command[2]-1])
return answer
하지만 map 과 lambda 함수를 사용해서 간단히 작성하는 방법도 존재한다.
def solution(array, commands):
return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))
'PS > 프로그래머스' 카테고리의 다른 글
체육복 (lv1) (0) | 2022.07.03 |
---|---|
모의고사 (lv1) (0) | 2022.07.03 |
완주하지 못한 선수 (lv1) (0) | 2022.07.03 |
소수 만들기 (lv1) (0) | 2022.07.03 |
키패드 누르기 (lv1) (0) | 2022.07.02 |