kth-smallest-element-in-a-sorted-matrix.py

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        return sorted([x for row in matrix for x in row])[k-1]