LeedCode_有序矩阵中第K小的元素

题目说明

给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。
请注意,它是排序后的第k小元素,而不是第k个元素。

matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。

链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix
分析:采用优先对列的方法,一定要注意,一定要遍历完所有元素

int kthSmallest(vector<vector<int>>& matrix, int k) 
{
	int n = matrix.size();
	if (n == 0)
	{
		return 0;
	}
	priority_queue<int> pq;
	for (int i = 0; i < matrix.size(); i++)
	{
		for (int j = 0; j < matrix[i].size(); j++)
		{
			pq.push(matrix[i][j]);
			if (pq.size() > k)
			{
				pq.pop();							//采用优先队列,一定要将矩阵中的所有元素遍历一遍
			}
		}
	}
	return pq.top();
}
发布了63 篇原创文章 · 获赞 14 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/luncy_yuan/article/details/104116683
今日推荐