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

有序矩阵中第k小的元素

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

示例:

matrix = [

[ 1, 5, 9],

[10, 11, 13],

[12, 13, 15]

],

k = 8,

 

返回 13。

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

 

根据二分搜索法，获取中间值，然后搜索他是否为第k个值。

 

class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        int len = matrix.length;
        int low = matrix[0][0],high= matrix[len-1][len-1];
        while(low<=high){
            int mid = low + (high-low)/2;
            int count = helper(matrix,mid);
            if(count<k) low = mid+1;
            else high = mid-1; //排除mid不在矩阵内的情况，所以只能等到low和high时才退出循环
        }
        return low;
    }
    private static int helper(int[][] matrix, int mid) {
        int i = matrix.length-1,j=0;
        int res = 0;
        while(i>=0&&j<matrix[0].length){
            if(matrix[i][j]>mid) i--;
            else{
                res+=i+1;
                j++;
            }
        }
        return res;
    }
}