PS/프로그래머스
k번째 수 (lv1)
ForteQook
2022. 7. 3. 16:03
배열의 원하는 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))