LeetCode 有序矩阵中第K小的元素（C++）

最新推荐文章于 2022-01-04 22:27:19 发布

shower_v

最新推荐文章于 2022-01-04 22:27:19 发布

阅读量513

点赞数

分类专栏： C++ leetcode

本文链接：https://blog.csdn.net/weixin_40349531/article/details/89160023

版权

leetcode 同时被 2 个专栏收录

83 篇文章 0 订阅

订阅专栏

C++

54 篇文章 0 订阅

订阅专栏

题目

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

示例:

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

返回 13。

说明:

你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。

解题思路：

维护一个最大堆，最大堆的size始终保持k，遍历一遍这个矩阵，然后返回这个最大堆的堆顶即可得到第K小的元素。
具体代码如下：

class Solution 
{
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) 
    {
        int first_index = matrix.size();
        int second_index = matrix[0].size();
        vector<int> mylist{};
        for (int i = 0; i < first_index; i++)
        {
            for (int j = 0; j < second_index; j++)
            {
                if (mylist.empty())
                {
                    mylist.push_back(matrix[i][j]);
                }
                else
                {
                    mylist.push_back(matrix[i][j]);
                    push_heap(mylist.begin(), mylist.end(), less<int>());
                    if (mylist.size() > k)
                    {
                        pop_heap(mylist.begin(), mylist.end(), less<int>());
                        mylist.pop_back();
                    }
                }
            }
        }
        return mylist[0];
    }
};