【日常训练】378. 有序矩阵中第 K 小的元素

代码

给你一个 n x n 矩阵 matrix ，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是 排序后 的第 k 小元素，而不是第 k 个 不同 的元素。
你必须找到一个内存复杂度优于 O(n²) 的解决方案。

示例 1：
输入：matrix = [[1,5,9],[10,11,13],[12,13,15]], k = 8
输出：13
解释：矩阵中的元素为 [1,5,9,10,11,12,13,13,15]，第 8 小元素是 13

示例 2：
输入：matrix = [[-5]], k = 1
输出：-5

提示：
n== matrix.length
n == matrix[i].length
1 <= n <= 300
-10⁹ <= matrix[i][j] <= 10⁹
题目数据 保证 matrix 中的所有行和列都按 非递减顺序 排列
1 <= k <= n²

代码

package leetcodeday1;

import java.util.ArrayList;
import java.util.Arrays;

public class leetcode378 {
    // 暴力，排序
//    public int kthSmallest(int[][] matrix, int k) {
//        int[] tmp = new int[matrix.length*matrix.length];
//        int t = 0;
//        for (int i = 0; i < matrix.length; i++){
//            for (int j = 0; j < matrix.length; j++){
//                tmp[t++] = matrix[i][j];
//            }
//        }
//        Arrays.sort(tmp);
//        return tmp[k - 1];
//    }

    // 二分查找
    public int kthSmallest(int[][] matrix, int k) {
        int n = matrix.length;
        int l = matrix[0][0];
        int r = matrix[n - 1][n - 1];
        while (l < r){
            int mid = l + ((r - l) >> 1);
            // 判断小于mid数有多少，如果大于k，则说明要找的数在这些数当中，即在矩阵左上方
            // 如果小于k，则在矩阵右下方
            if (checkNums(matrix, mid, n , k)){
                r = mid;
            } else{
                l = mid + 1;
            }
        }
        return l;
    }

    // 统计小于mid数字的个数
    public boolean checkNums(int[][] matrix, int mid, int n, int k) {
        // 从左下角开始走，先向右走，走不通之后向上走，统计途径数字的个数
        int i = n - 1;
        int j = 0;
        int cnt = 0;
        while (i >= 0 && j < n ){
            if (matrix[i][j] <= mid){
                // 一次加入一列数据
                cnt += i + 1;
                j++;
            }else {
                // 向矩阵上方移动
                i--;
            }
        }
        return cnt>=k;
    }

    public static void main(String[] args) {
        leetcode378 obj = new leetcode378();
        int[][] matrix ={
                {-5,-4},{-5,-4}
        };
        System.out.println(obj.kthSmallest(matrix, 2));
    }
}