COS Pro 2급 Python 모의고사 - [YBM] k번째로 작은 수

문제 설명

 

자연수가 담겨있는 n x 4 크기의 2차원 배열에서 k번째로 작은 수를 찾으려 합니다. 이때, n은 배열의 세로길이, 4는 배열의 가로길이입니다. 예를 들어 다음은 자연수가 담겨있는 4 x 4 크기의 2차원 배열입니다.

위 2차원 배열에서 가장 작은 수는 2입니다. 두 번째로 작은 수는 4, 세 번째로 작은 수는 5이며, 네 번째로 작은 수는 11입니다.

2차원 배열 arr와 k가 매개변수로 주어질 때, arr에서 k번째로 작은 수를 찾아 return 하도록 solution 함수를 완성해주세요.

---

매개변수 설명

 

2차원 배열 arr와 자연수 k가 solution 함수의 매개변수로 주어집니다.

• arr는 n x 4 크기의 2차원 배열이며, n은 1 이상 20 이하입니다.
• n은 arr의 세로길이이며, 가로길이는 항상 4입니다.
• arr의 원소는 1 이상 1,000 이하의 자연수이며, 같은 수가 중복해서 들어있지 않습니다.
• k는 1 이상 n x 4 이하의 자연수입니다.

---

return 값 설명

2차원 배열 arr에서 k번째로 작은 수를 찾아서 return 하도록 solution 함수를 작성해주세요.

---

예시

arr	k	return
[[5,12,4,31],[24,13,11,2],[43,44,19,26],[33,65,20,21]]	4	11

예시 설명

 

예시 #1
주어진 2차원 배열은 문제의 그림과 같습니다. 이때, 네 번째로 작은 수는 11입니다.

 

 

---

def solution(arr, k):
    list = []
    for i in arr:
        list += i
    list.sort()
    return list[k-1]
print(solution([[5,12,4,31],[24,13,11,2],[43,44,19,26],[33,65,20,21]],4))