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

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

示例:

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

返回 13。

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

-------------------------------------------------------------------------------------------------------------------------------

如果用STL中的库函数的话,那么可以写成

  1. 两个迭代器相减代表的是两个之间的距离。
  2. lower_bound() 返回一个迭代器,指向第一个不小于目标val的元素。
  3. upper_bound()返回一个迭代器,指向第一个大于目标val的元素。
class Solution {
public:

    int h,w;

    int count(vector<vector<int>>& matrix, int t){
        int i = h - 1;
        int j = 0,sum = 0;
        while(i >= 0 && j <= w - 1){
            if (matrix[i][j] <= t){
                ++j;
                sum += i + 1;
            }else{
                --i;
            }
        }
        return sum;
    }

    int kthSmallest(vector<vector<int>>& matrix, int k) {
        h = matrix.size();
        w = matrix[0].size();
        int l = matrix[0][0];
        int r = matrix[h - 1][w - 1];
        while(l < r){                    // 注意这里二分的写法,是求二分的左边界
            int mid = l + r >> 1;
            if (count(matrix, mid) >= k){
                r = mid;
            }else{
                l = mid + 1;
            }
        }
        return l;
    }
};

问题一:

怎么确定函数返回的l就一定在矩阵中呢?

就拿题目中的例子来说,单独看13和14都满足count()结果为8,但是应该返回的是13。为什么呢? 因为如果14是存在于数组中的,13的count()就不可能是8,而应该是7,因为此时比13大的数有14、15。总结下:[13,15)之间的数是有着同一个count()的,是原因13在数组中,而(13,15)之间的数又不在数组中。

问题二:

count()函数的工作方式是什么样的?

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值