401.Kth Smallest Number in Sorted Matrix-排序矩阵中的从小到大第k个数（中等题）

排序矩阵中的从小到大第k个数

1. 题目

   在一个排序矩阵中找从小到大的第 k 个整数。
   排序矩阵的定义为：每一行递增，每一列也递增。

2. 样例

   给出 k = 4 和一个排序矩阵：
   

3. 挑战

   使用O(k log n)的方法，n为矩阵的宽度和高度中的最大值。

4. 题解

   使用优先队列进行最小堆排序，每次选取最小的元素，第K次选择的就是第K大元素。

public class Solution {    class Pair     {        public int x, y, val;        public Pair(int x, int y, int val)         {            this.x = x;            this.y = y;            this.val = val;        }    }    /**     * @param matrix: a matrix of integers     * @param k: an integer     * @return: the kth smallest number in the matrix     */    public int kthSmallest(int[][] matrix, int k) {        int[] dx = new int[]{0, 1};        int[] dy = new int[]{1, 0};        int n = matrix.length;        int m = matrix[0].length;        boolean[][] hash = new boolean[n][m];        PriorityQueue<Pair> minHeap = new PriorityQueue<Pair>(k, new Comparator<Pair>()        {            public int compare(Pair a, Pair b)             {                return a.val - b.val;            }        });        minHeap.add(new Pair(0, 0, matrix[0][0]));        for(int i = 0; i < k - 1; i++)        {            Pair cur = minHeap.poll();            for(int j = 0; j < 2; j++)            {                int next_x = cur.x + dx[j];                int next_y = cur.y + dy[j];                if(next_x < n && next_y < m && !hash[next_x][next_y])                {                    hash[next_x][next_y] = true;                    Pair next_Pair = new Pair(next_x, next_y, matrix[next_x][next_y]);                    minHeap.add(next_Pair);                }            }        }        return minHeap.peek().val;    }}

Last Update 2016.11.15