코딩테스트 연습 - k번째 수 문제 풀기

오늘 리뷰할 코딩테스트 문제는 k번째에 있는 수를 구하는 문제

 

우선 문제부터

 

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, 
k번째에 있는 수를 구하려 합니다.

예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면

array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다.
1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다.
2에서 나온 배열의 3번째 숫자는 5입니다.
배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, 
commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아
return 하도록 solution 함수를 작성해주세요.

 

아래는 내가 작성한 코드

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

 

 

코드 풀이를 하자면

 

 

위의 이미지가 매개변수로 입력되는 값이고, return 값처럼 출력을 하도록 만들어야한다.

for문을 통해 commands의 각 원소를 순회하고, 슬라이싱을 통해 시작점, 끝점, 찾아야 하는 위치를 각각

변수에 할당을 했으며,

매개변수 array의 값을 시작점 i, 끝점 j, 찾아야 하는 위치 k를 활용해 슬라이싱 및 정렬 후 값을 answer에

추가하는 방식으로 문제 풀이

 

코드의 성능

• 메모리 10.3 MB
• 시간 0.01 ms