문제

프로그래머스 연습 문제 Lv. 1

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.
배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.
예를 들어 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입니다.

나의 답변

def solution(array, commands):
    answer = []
    result = []
    for i in range(len(commands)):
        answer = array[commands[i][0]-1: commands[i][1]] # array[2:5] => array[1:5] answer = [5, 2, 6, 3]
        answer.sort()
        result.append(answer[commands[i][2]-1])
    return result

• 매개변수인 commands가 2차원 배열이기 때문에, for문에서 i를 commands의 요소의 수만큼 반복하도록 했다.

• 이후 문제에 나온 각 숫자 i, j, k는 인덱싱을 사용한 commands[i][0], commands[i][1], commands[i][2] 을 통해 표현하고자 했다.

• 우선 array에서 i부터 j까지 자르기 위해 슬라이싱을 사용했다.
  파이썬에서 인덱스는 0부터 시작하기 때문에, 내가 원하는 것인 i번째 숫자를 구하기 위해서는, commands[i]의 0번째 인덱스에서 1을 빼야 한다.

• 따라서 array를 i부터 j까지 자른 것을 array[commands[i][0]-1: commands[i][1]] 이라 표현했고, 이것을 answer 변수에 저장했다.
  이때 answer는 array의 i번째부터 j번째까지의 요소로 구성된 리스트가 된다.

• 이후 정렬을 위해 answer.sort() 메서드를 사용했다.

• 정렬 후 k번째 숫자를 구하기 위해, commands[i][2]를 이용했다.
  마찬가지로 인덱스가 0부터 시작하는 원리를 따라 인덱스에서 1을 뺀, answer[commands[i][2]-1]을 사용하여 result 변수에 담았다.

---

다른 사람의 답변

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

• for문에서 반복변수 command를 언팩킹(unpacking)해서 각각 i, j, k라는 변수에 담았다. command가 [i, j, k] 형태의 리스트일 때 변수 i, j, k에 각각의 값이 할당된다.
  리스트나 튜플의 요소를 여러 변수에 동시에 할당하는 방법이다.

• 이 방법은 for i, j, k in commands: 로도 활용할 수 있다.

• 또한 sorted()결과는 리스트라서 list(sorted(... 중 list 함수는 제거가 가능하다는 피드백이 있었다.

def solution(array, commands):
    return [sorted(array[a[0]-1:a[1]])[a[2]-1] for a in commands]

• 리스트 컴프리헨션 방법을 사용했다.

• 반복문인 for a in commands를 맨 뒤에 적고, array[a[0]-1:a[1]] 을 통해 i부터 j까지의 배열을 뽑아내 sorted( )를 사용해 정렬했다. 이후 정렬한 리스트에서 [a[2]-1] 인덱싱을 사용해 k번째 수를 뽑아냈다.