파이썬/알고리즘 공부

[python] 프로그래머스 K번째 수 🖖

Jimyeung 2021. 10. 28. 20:19
728x90

K번째수

❓문제 설명

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

예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3일경우

1. array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다.
2. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다.
3. 2에서 나온 배열의 3번째 숫자는 5입니다.

배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

✅제한사항

  • array의 길이는 1 이상 100 이하입니다.
  • array의 각 원소는 1 이상 100 이하입니다.
  • commands의 길이는 1 이상 50 이하입니다.
  • commands의 각 원소는 길이가 3입니다.

✅입출력 예

array commands return
[1, 5, 2, 6, 3, 7, 4] [ [2, 5, 3], [4, 4, 1], [1, 7, 3] ] [5, 6, 3]

✅입출력 예 설명
[1, 5, 2, 6, 3, 7, 4]를 2번째부터 5번째까지 자른 후 정렬합니다. [2, 3, 5, 6]의 세 번째 숫자는 5입니다.

[1, 5, 2, 6, 3, 7, 4]를 4번째부터 4번째까지 자른 후 정렬합니다. [6]의 첫 번째 숫자는 6입니다.

[1, 5, 2, 6, 3, 7, 4]를 1번째부터 7번째까지 자릅니다. [1, 2, 3, 4, 5, 6, 7]의 세 번째 숫자는 3입니다.


💢이해를 돕기위한 세분화

a에 commands에 담긴 각 리스트들을 담아준다

for a in commands:

>>>a1리스트[2, 5, 3]
>>>a2리스트[4, 4, 1]
>>>a3리스트[5, 6, 3]

✅ 각 리스트에 담긴 첫번째값부터 2번째 값까지의 수들을 lst라는 리수트 변수에 담아준다

※인덱스는 0부터 시작하므로 시작부분의 -1을 해준다.

array = [ 1, 5, 2, 6, 3, 7, 4]
lst = array[a[0]-1:a[1]]

>>>a = [2, 5, 3]일 경우 => [5, 2, 6, 3] (2부터 5까지)
>>>a = [4, 4, 1]일 경우 => [6] (4부터4까지)
>>>a = [1, 7, 3]일 경우 => [1, 5, 2, 6, 3, 7, 4] (1부터 7까지)

✅ 뽑아낸 숫자들을 정렬해준다.

lst.sort()
>>>[2, 3, 5, 6]
>>>[6] 
>>>[1, 2, 3, 4, 5, 6, 7]

✅정렬된 숫자들을 result리스트 변수에 담아준다.(마찬가지로 인덱스는 0부터 시작하므로 -1을 해 주어야 한다.)

result.append(lst[a[2]-1])

>>>[5, 6, 3]

최종 코드

def solution(array, commands):
    result = []
    for a in commands: 
        lst = array[a[0]-1:a[1]]
        lst.sort()
        result.append(lst[a[2]-1])
    print(result)
    return result

array = [1, 5, 2, 6, 3, 7, 4]
commands = [[2, 5, 3], [4, 4, 1], [1, 7, 3]]
solution(array, commands)