[프로그래머스] k번째수

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

프로그래머스

---

처음 작성한 코드 (정답)

def solution(array, commands):
    answer = []
    for command in commands:
        i, j, k = command
        temp = []
        for x in range(i-1,j):
            temp.append(array[x])
        temp.sort()
        answer.append(temp[k-1])
    return answer

 

수정한 코드 (정답)

def solution(array, commands):
    answer = []
    for command in commands:
        i, j, k = command
        temp = sorted(array[i-1:j])
        answer.append(temp[k-1])
    return answer

리스트 슬라이싱을 하면 불필요한 for문 사용을 하지 않아도 되었다.

 

---

사소한 거지만, 당연히 temp 변수를 생성하지 않고 아래와 같이 한 줄로 써줄 수도 있다.

#한줄
answer.append( sorted(array[i-1:j])[k-1] )

#두줄
temp = sorted(array[i-1:j])
answer.append(temp[k-1])

개인적으로 한줄짜리는 가독성이 좀 떨어지는 것 같아 두 줄로 분리해서 작성했다.