Leetcode.0378 | 有序矩阵中第k小的元素

题目

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

示例

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

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

解决方法

使用二分法，定义一个mid作为当前区间左右两端的平均值。然后计算出矩阵中小于等于mid的元素的个数count。

1. 如果count>=k，则说明矩阵中，小于等于mid的元素>=k个，要找的第k小的元素位于mid的左侧。那么，就应该将区间的右端缩到mid，继续迭代进行查找。

2. 如果count<k，则说明矩阵中，小于等于mid的元素小于k个，要找的第k小的元素位于mid的右侧。那么，就应该将区间的左端变到mid+1，继续迭代进行查找。

循环结束时，left和right值相等，返回其中之一。

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        int row = matrix.size();//行数
        int col = matrix[0].size();//列数
        int min = matrix[0][0];//区间的左边界
        int max = matrix[row-1][col-1];//区间的右边界
        while(min < max){
            int mid = min + (max - min) / 2;
            int count = 0;
            if(judgeLeft(matrix,mid) >= k){
                max = mid;
            }else{
                min = mid + 1;
            }
        }
        return max;
    }
    int judgeLeft(vector<vector<int>>& matrix,int mid){
        int row = matrix.size(); //行数
        int col = matrix[0].size(); //列数
        int i = row-1; //起始点的横坐标
        int j = 0; //起始点的纵坐标
        //从左下角到右上角进行查找
        int count = 0;
        while(i >= 0 && j <= col-1){
                if(matrix[i][j] <= mid) {
                // 说明matrix[i][j]及上边的元素都<=mid
                // 将i及以上的数量加进count中
                count += (i + 1);
                // 判断右边列
                j++;
            }else {
                // i向上移动
                i--;
            }
        }
        return count;
    }
};