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

题目：给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是排序后的第 k 小元素，而不是第 k 个不同的元素。
思路：最小值 matix[0][0]，最大值 matrix[n-1][n-1]，可以利用二分去寻找“小于等于 val 的数量大于等于 k” 的 mid val，而计算这个数量可以参考 240. 搜索二维矩阵 II，从左下角开始统计某列小于等于 val 的数目。
代码：

class Solution {
public:
    bool check(vector<vector<int>>& matrix, int k, int target) {
        int n = matrix.size();
        int i = n - 1, j = 0, sum = 0;
        while (i >= 0 && j < n) {
            if (matrix[i][j] <= target) {
                sum += i + 1;
                ++j;
            } else {
                --i;
            }
        }
        return sum >= k;
    }
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        int n = matrix.size();
        if (n == 0) {
            return 0;
        }
        int left = matrix[0][0], right = matrix[n-1][n-1];
        while (left < right) {
            int mid = left + (right - left) / 2;
            if (check(matrix, k, mid)) {
                right = mid;
            } else {
                left = mid + 1;
            }
        }
        return left;
    }
};