【堆】leetcode378.有序矩阵中第K小的元素

题目：
给你一个 n x n 矩阵 matrix ，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是 排序后 的第 k 小元素，而不是第 k 个 不同 的元素。

思路：
求第K小，使用大根堆，堆顶元素即为第K小

解答：

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        #堆排序，大根堆的堆顶元素为最大的
        #比堆顶元素小，则将堆顶元素弹出，将当前元素入堆
        n=len(matrix)
        count=0
        heap=[]
        for i in range(n):
            for j in range(n):
                if count<k:
                    heapq.heappush(heap,-matrix[i][j])
                    count+=1
                else:
                    if -matrix[i][j]>heap[0]:
                        heapq.heappop(heap)
                        heapq.heappush(heap,-matrix[i][j])
        return -heap[0]