给定一个 的矩阵,每行每列都是升序排列的,请你找到矩阵中第 K 小的元素
数据范围:
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param matrix int整型ArrayList<ArrayList<>> * @param k int整型 * @return int整型 */ public int KthinMatrix(ArrayList<ArrayList<Integer>> matrix, int k) { // write code here int row = matrix.size(); int col = matrix.get(0).size(); Queue<Integer> queue = new PriorityQueue<>(); for (int i = 0; i < row; i++) { for (int j = 0; j < col; j++) { queue.add(matrix.get(i).get(j)); } } while (k > 1) { queue.poll(); k--; } return queue.poll(); } }