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

[抄题]:

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

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

[思维问题]:

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

[一句话思路]:

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

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

[画图]:

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

[一刷]:

  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;
}
}
上一篇:vue实例讲解之axios的使用


下一篇:HDU4758 Walk Through Squares AC自动机&&dp