378. Kth Smallest Element in a Sorted Matrix

문제 링크 : https://leetcode.com/problems/kth-smallest-element-in-a-sorted-matrix/

주어진 매트릭스에서 k번째로 큰 수를 찾는 문제이다.

단순히 주어진 배열에서 n번째로 큰 수를 찾는건 어렵지않으나 2차원배열이기때문에 난이도가 medium이 되지 않았나싶다.

그래서 새로운 1차원배열을 만들어서 append하여 쉽게 풀었다.

그래서 처음에 짠 코드

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        m = []
        for i in matrix:
            for j in i:
                m.append(j)
        return m[k-1]

그리고 실패한 테.케

Input:
[[1,2],[1,3]]
2
Output:
2
Expected:
1

(니들이 먼저 sorted되어있다 했잖아 )

그래서 일차원배열 m을 다시 sort해주었더니 성공

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        m = []
        for i in matrix:
            for j in i:
                m.append(j)
        m.sort()
        return m[k-1]