排序矩阵中的从小到大第k个数 · Kth Smallest Number In Sorted Matrix

[抄题]:

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

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

[思维问题]:

不知道应该怎么加,因为不是一维单调的。

[一句话思路]:

周围两个数x或y挪一位, 如果hash数组没有就添加到minheap中

[输入量]:空: 正常情况:特大:特小:程序里处理到的特殊情况:异常情况(不合法不合理的输入):

[画图]:

 

[一刷]:

  1. class Pair(没有参数)中要有数据类型、方法Pair(int x , int y, int val)
  2. PairComparator 继承了 Comparator<Pair>类 里面自然就是对pair进行比较, 接口<内容物>+名 = new 内部结构名<内容物>。Queue xx = new PQ<Integer>(k,新建pair对象) 0,0,matrix[0][0] 也要新建对象 new Pair
  3. x坐标用dx表示, 里面只有0,1
  4. 取出数从i = 1开始,不用从0开始。因为第一个符合条件的pair是(0,0,matrix[0][0])。
  5. 数组不包括,直接用!hash[x][y]即可

[二刷]:

  1. class关键字是小写开头
  2. next_pair新建之后,minheap直接add(next_pair)即可
  3. dx,dy数组的作用是加一位坐标,因此里面有东西:0 或 1 

[三刷]:

  1. paircomparator类是继承来的,别的地方可以用,所以写class

[四刷]:

[五刷]:

[总结]:

[复杂度]:Time complexity: O(klgk) Space complexity: O(k)

[英文数据结构,为什么不用别的数据结构]:

heap符合每次取顶 丢掉,取到k个为止

[其他解法]:

二分法

[Follow Up]:

[LC给出的题目变变变]:

373. Find K Pairs with Smallest Sums 也是用minheap实现取最小

668. Kth Smallest Number in Multiplication Table 看着像用heap,其实用二分法 玩数学

719. Find K-th Smallest Pair Distance 其实用二分法 玩数学。看来二分法也能用于查找最小的第k个数

public class Solution {
    /*
     * @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) {
        class Pair {
            int x,y,val;
            public Pair(int x,int y,int val) {
                this.x = x;
                this.y = y;
                this.val = val;
            }
        };
        class PairComparator implements Comparator<Pair>{
            public int compare(Pair a,Pair b) {
                return a.val - b.val;
            }
        };
        int m = matrix.length;
        int n = matrix[0].length;
        int[] dx = new int[]{0,1};
        int[] dy = new int[]{1,0};
        boolean[][] hash = new boolean[m][n];
        Queue<Pair> minHeap = new PriorityQueue<Pair>(k,new PairComparator());
        minHeap.add(new Pair(0,0,matrix[0][0]));
        for (int i = 1; i < k; 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];
                Pair next_Pair = new Pair(next_x,next_y,0);
                if (next_x < m && next_y < n && !hash[next_x][next_y]) {
                    next_Pair.val = matrix[next_x][next_y];
                    hash[next_x][next_y] = true;
                    minHeap.add(next_Pair);
                }
            }
        }
        return minHeap.peek().val;
    }
}
View Code

 

posted @ 2018-01-19 11:48  苗妙苗  阅读(193)  评论(0编辑  收藏  举报