[프로그래머스 python] 정렬 - K번째 수

이혜지·2022년 2월 1일
0

Algorithm

목록 보기
4/12
post-thumbnail

해당 문제 아래 링크 확인 가능
https://programmers.co.kr/learn/courses/30/lessons/42748

문제 풀이

def solution(array, commands):
    answer = []
    tmp = []

    for x in range(len(commands)):
        i = commands[x][0]
        j = commands[x][1]
        k = commands[x][2]
        tmp = array[i - 1 : j]
        tmp.sort()
        answer.append(tmp[k - 1])
    return answer

이렇게 풀었었는데 쓸데없이 길어서 아래로 고쳤다.

문제 풀이2

def solution(array, commands):
    answer = []
    
    for command in commands:
        arr = array[command[0] - 1:command[1]] #sorted 자체는 납두고 새로운 배열을 생성하고 정렬 후 반환, #sort 
        arr.sort()
        answer.append(arr[command[2] - 1])
    return answer

profile
공유 문화를 지향하는 개발자입니다.

0개의 댓글