LeetCode—有序矩陣中第k小的元素（排序+官方二分）

有序矩陣中第k小的元素（中等）

2020年7月2日

題目來源：力扣

力扣

• 排序

試驗的做法，二維轉一維再排序，花時間費空間

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

• 二分

學習官方的二分題解，原來二分還能這麼用，學到了學到了

class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        //二維數組的長度，因爲是n*n，所以算一個就行
        int n = matrix.length;
        //左邊界
        int left = matrix[0][0];
        //右邊界
        int right = matrix[n - 1][n - 1];
        //使用二分，這裏的>>1就是二進制的除以2
        while (left < right) {
            int mid = left + ((right - left) >> 1);
            if (check(matrix, mid, k, n)) {
                right = mid;
            } else {
                left = mid + 1;
            }
        }
        return left;
    }
    //從左下角開始，符合小於等於mid就右移，不符合就上移，記錄這一列小於mid的數量
    public boolean check(int[][] matrix, int mid, int k, int n) {
        int i = n - 1;
        int j = 0;
        int num = 0;
        while (i >= 0 && j < n) {
            if (matrix[i][j] <= mid) {
                num += i + 1;
                j++;
            } else {
                i--;
            }
        }
        return num >= k;
    }
}