[프로그래머스 / Python] K번째수(정렬) - Lv.1

K번째수(정렬) - Lv.1

---

문제 설명

배열 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입니다.

---

풀이

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

 

---

다른 사람의 풀이

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

나랑 비슷한 풀이인데 나처럼 굳이 command에 하나씩 지정하지 않더라도 i, j, k가 순서대로 들어간다.

또한 댓글을 보니까 // "for command in commands: i,j,k = command" 대신 "for i,j,k in commands:" 로 한 줄로 줄여 쓰면 더 편해요! // 를 보고 또 하나 배웠다.

 

def solution(array, commands):
    return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))

이건 풀이 중에서 가장 좋아요를 많이 받은 코드인데 람다 함수를 이용하여 한줄로 해결했는데 처음 보자마자 바로 이해가 되지 않는다.. 저번에 람다 함수에 대해서 블로그에 정리하긴 했었는데 이런 식으로 lambda와 map 함수를 같이 활용하는 건 처음 본다. 공부해보면 다음에 정말 유용할 것 같다.