배열의 원하는 index를 잘라서 가져오는 정도의 간단한 문제이다.
Lambda 함수
3.5 람다(lambda) - 왕초보를 위한 Python: 쉽게 풀어 쓴 기초 문법과 실습 (wikidocs.net)
3.5 람다(lambda)
오늘은 람다 형식과 그것을 이용하는 여러 가지 함수들에 대해서 알아보겠습니다. 당장 완벽하게 소화하실 필요는 없을 것 같구요, 가벼운 마음으로 이런 것이 있다는 정도만 아셔 ...
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 |