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

在一个排序矩阵中找从小到大的第 k 个整数。

排序矩阵的定义为：每一行递增，每一列也递增。

您在真实的面试中是否遇到过这个题？

Yes





样例

给出 k = 4 和一个排序矩阵：
[
  [1 ,5 ,7],
  [3 ,7 ,8],
  [4 ,8 ,9],
]


返回 5。

挑战

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



相关题目 Expand   

解题思路：

可以利用java自带的最小堆来进行每次选取最小的元素，一共选择K次就行了。

public class Solution {
     /**
     * @param matrix
     *            : a matrix of integers
     * @param k
     *            : an integer
     * @return: the kth smallest number in the matrix
     */
     class S {
          int row;
          int col;
          int val;
          public S(int row, int col, int val) {
               super();
               this.row = row;
               this.col = col;
               this.val = val;
          }

     }
     public int kthSmallest(int[][] matrix, int k) {
          // write your code here
          int m = matrix.length;
          if (0 == m)
               return -1;
          int n = matrix[0].length;
          PriorityQueue<S> queue = new PriorityQueue<S>(n,new MyComparators());
          for(int i=0;i<n;i++){
               queue.add(new S(0, i, matrix[0][i]));
          }
          for(int i=0;i<k-1;i++){
               S tmp = queue.poll();
               if(tmp.row<m-1){
                    queue.add(new S(tmp.row+1, tmp.col,matrix[tmp.row+1][tmp.col]));
               }              
          }
          return queue.poll().val;
     }
     @SuppressWarnings("unchecked") 
     class MyComparators implements Comparator<S> {
          @Override
          public int compare(S o1, S o2) {
               // TODO Auto-generated method stub
               if (o1.val > o2.val) {
                    return 1;
               } else if (o1.val == o2.val) {
                    return 0;
               } else {
                    return -1;
               }
          }

     }
}