378. 有序矩阵中第K小的元素/C++

原创于 2019-07-25 19:45:23 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode/C++ 专栏收录该内容

248 篇文章

订阅专栏

本文介绍了一种高效的算法，用于在二维矩阵中寻找第K小的元素。该算法利用了二分查找的思想，首先确定搜索范围，然后通过计算矩阵中小于等于中间值的元素数量来逐步缩小范围，直至找到目标元素。

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

在这里插入图片描述

class Solution {
public:
	//找到<=k的元素的个数
    int search(vector<int>& vec, int k){
        int l = 0, r = vec.size();
        while(l<r){
            int m = l + (r - l)/2;
            if(vec[m]>k)
                r=m;
            else
                l=m+1;
        }
        return l;
    }
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        int lv = matrix[0][0], rv = matrix.back().back();
        while(lv<rv){
            int mv = lv + (rv - lv)/2;
            int cnt = 0;
            for(int i=0;i<matrix.size();++i)
                cnt+=search(matrix[i],mv);
            //cnt是小于等于mv的元素总个数
            if(cnt>=k)
                rv = mv;
            else
                lv = mv+1;
        }
        return lv;
    }
};