leetcode:有序矩阵中第K小的元素(二分查找)

https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix/

暴力法：

直接将矩阵中的数，全部存到一个数组中，然后排序。（竟然没超时）

class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        int n = matrix.length;
        int[] a = new int[n*n];
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                a[cnt++] = matrix[i][j];
            }
        }
        Arrays.sort(a);
        return a[k-1];
    }
}

二分查找

因为这个矩阵是有序的，左上角的数，大于右下角的数。可以用到二分查找的方法。我们先用left和right分别记录矩阵中最小和最大的数，令mid = (left + right)，可以把矩阵分为左上和右下两个部分，然后统计矩阵中大于等于mid的数cnt

如果 cnt < k,则表明第k个数在右下部分，否则第k个数就在左上部分。然后一直找下去，直到left==right， 返回left的值。

这里有个计算cnt的技巧：

因为矩阵从上往下是递增的，我们可以选择从左下往右上找（或者右上往左下），比如说左下的一个数小于mid，那么，它的这列的数字从上到这个数都是小于mid的，这样可以更方便的统计。

class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        int n = matrix.length;
        int left = matrix[0][0];
        int right = matrix[n-1][n-1];
        while (left < right){
            int mid = (left + right)/2;
            int cnt = count(matrix,mid,n);
            if(cnt < k){
                left = mid + 1;
            }else{
                right = mid;//可能包含mid
            }
        }
        return left;
    }
    //计算矩阵中 不大于mid的个数,从左下开始找
    int count(int[][] matrix, int mid, int n) {
        int i = n -1 ,j = 0,cnt = 0;
        while (i>=0 && j < n){
            if(matrix[i][j] <= mid){
                cnt+=i+1;//这一列到i的个数
                j++;
            }else{
                i--;
            }
        }
        return cnt;
    }
}

太难了~~