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))

 

'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