​LeetCode刷题实战378:有序矩阵中第 K 小的元素

本文介绍了如何解决LeetCode上的第338题——有序矩阵中第K小的元素。提供了两种解决方案,一种是使用优先队列,另一种是运用二分查找思想。通过这两种方法,可以在有序矩阵中有效地找到第K小的元素。文章还给出了详细的代码实现,并附带了示例解释。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !

今天和大家聊的问题叫做 有序矩阵中第 K 小的元素,我们先来看题面:

https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix/

Given an n x n matrix where each of the rows and columns are sorted in ascending order, return the kth smallest element in the matrix.

Note that it is the kth smallest element in the sorted order, not the kth distinct element.

给你一个 n x n 矩阵 matrix ,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。

请注意,它是 排序后 的第 k 小元素,而不是第 k 个 不同 的元素。

示例

示例 1:

输入:matrix = [[1,5,9],[10,11,13],[12,13,15]], k = 8
输出:13
解释:矩阵中的元素为 [1,5,9,10,11,12,13,13,15],第 8 小元素是 13

示例 2:

输入:matrix = [[-5]], k = 1
输出:-5

解题

1)利用优先队列进行处理

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        priority_queue<int> pq;
        for(int i=0;i<matrix.size();i++){
            for(int j=0;j<matrix[0].size();j++){
                pq.emplace(matrix[i][j]);
                if(pq.size()>k) pq.pop();
            }
        }
        return pq.top();
    }
};

2)利用二分思路进行处理

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        int left=matrix[0][0],right=matrix.back().back();
        while(left<right){
            int mid=left+(right-left)/2,cnt=0;
            for(int i=0;i<matrix.size();i++){
                cnt+=upper_bound(matrix[i].begin(),matrix[i].end(),mid)-matrix[i].begin();
            }
            if(cnt<k) left=mid+1;
            else right=mid;
        }
        return left;
    }
};

好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力 。

上期推文:

LeetCode1-360题汇总,希望对你有点帮助!

LeetCode刷题实战361:轰炸敌人

LeetCode刷题实战362:敲击计数器

LeetCode刷题实战363:矩形区域不超过 K 的最大数值和

LeetCode刷题实战364:加权嵌套序列和 II

LeetCode刷题实战365:水壶问题

LeetCode刷题实战366:寻找二叉树的叶子节点

LeetCode刷题实战367:有效的完全平方数

LeetCode刷题实战368:最大整除子集数

LeetCode刷题实战369:给单链表加一

LeetCode刷题实战370:区间加法

LeetCode刷题实战371:两整数之和

LeetCode刷题实战372:超级次方

LeetCode刷题实战373:查找和最小的K对数字

LeetCode刷题实战374:猜数字大小

LeetCode刷题实战375:猜数字大小 II

LeetCode刷题实战376:摆动序列

LeetCode刷题实战377:组合总和 Ⅳ

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

编程IT圈

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值