K번째 수

첫번째 코드

 

인덱싱이 중요한 문제라는 생각이 들었다!

commands의 길이만큼 반복할 것이지만, 해당 값들은 모두 인덱스가 1로 시작할 때 기준이므로 1을 빼고 인덱싱을 해주어야 한다.

해당 과정을 통해 새로운 리스트가 생성된다!!

 

sort를 통해 정렬을 진행하고

새로운 리스트에서 다시 인덱싱을 진행하여 정답에 추가한다!!

 

def solution(array, commands):
    answer = []
    
    for i in range(len(commands)):
        start = commands[i][0] - 1
        end = commands[i][1]
        new_array = array[start:end]
        new_array.sort()
        
        answer_index = commands[i][2] - 1
        answer.append(new_array[answer_index])
    return answer

 

통과!

---

링크

https://github.com/ornni/programmers/tree/main/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4/1/42748.%E2%80%85K%EB%B2%88%EC%A7%B8%EC%88%98

programmers/프로그래머스/1/42748. K번째수 at main · ornni/programmers