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;
     }
 }