leetcode | 378 Kth Smallest Element in a Sorted Matrix [Python]

💡 Python 3

🔗 문제

---

378. Kth Smallest Element in a Sorted Matrix [Link]

💻 코드

---

스터디 진도에 맞춰서 힙으로 풀었는데 더 좋은 방법이 있지 않을까 싶다.
그렇게 효율적인 코드는 아닌 것 같다는 생각이 스멀스멀 올라온다.
heapq.nsmallest를 한 번 써보고 싶었는데 이번에 적용해봤다.

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        heap = []

        for mat in matrix:
            for ele in mat:
                heapq.heappush(heap, ele)

        return heapq.nsmallest(k, heap)[-1]

참고 자료

---

heapq — 힙 큐 알고리즘

heapq모듈에 있는 nlargeest(), nsmallest() 함수