[프로그래머스/Python] Level1 K번째수 (정렬)

프로그래머스(Programmers) 코딩테스트 연습

Level1 정렬 K번째수 파이썬(Python) 풀이

• 풀이로 이동!

문제풀이

반복문과 slicing, sort 함수를 사용하면 간단하게 해결되는 가장 기초적인 정렬 문제이다.

1. 2차원 배열인 commands를 반복문을 통해 하나씩 접근한다.
2. 1차원 배열인 command에서 0번 index와 1번 index에 해당되도록 slicing 한다.
3. slicing된 배열을 정렬한다.
4. 정렬된 배열에서 index 2번에 해당되는 인자값을 answer 배열에 append 한다.
5. return answer

코드

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

if __name__ == "__main__":
    array = [1, 5, 2, 6, 3, 7, 4]
    commands = [[2, 5, 3], [4, 4, 1], [1, 7, 3]]
    print(solution(array, commands))