矩阵第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();
    }
}