Python/알고리즘

정렬 Level 1 K번째수 - python (프로그래머스)

구름용 2023. 4. 30. 22:25

https://school.programmers.co.kr/learn/courses/30/lessons/42748

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

소스 코드

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

 

풀이

문제에서 나온 흐름 그대로 진행하면 간단히 풀리는 문제였다. for 문을 통해 각각의 command 리스트를 조회하며 array 리스트를 나눈다음 정렬한 뒤 k번쨰 인덱스의 값을 answer 리스트에 추가하면 됐다.

 

다른 사람의 풀이

다른 사람의 풀이를 봐도 흐름은 같았지만 lambda를 사용하는 등 코드를 단축하여 표현하였다.