leetcode 378. 有序矩阵中第K小的元素

image.png

class Solution {
public:
    int kthSmallest(vector>& matrix, int k) {
//创建一个空最小堆
        priority_queue>,vector>>,greater>>> q;
        for(int i=0;i> temp = q.top();
            q.pop();//逐个pop，pop到第k-1 然后最小堆，堆顶就是result
            if(temp.second.second != matrix.size()-1){
                q.push(make_pair(matrix[temp.second.first][temp.second.second+1],make_pair(temp.second.first,temp.second.second+1)));
            }
            k--;
        }
        return q.top().first;
    }
};