LC378 - 有序矩阵中第K小的元素 - 优先队列/比较器

题目

有序矩阵中第K小的元素

matrix = [
   [ 1,  5,  9],
   [10, 11, 13],
   [12, 13, 15]
],
k = 8,

返回 13。

思路

方法一：直接排序

class Solution {
    public int kthSmallest(int[][] matrix, int k) {

        //方法1 直接排序
        int rows = matrix.length;
        int cols = matrix[0].length;
        int[] sortedList = new int[rows * cols];
        int index = 0;
        for(int[] row: matrix){
            for(int element: row){
                sortedList[index++] = element;
            }
        }
        Arrays.sort(sortedList);
        return sortedList[k-1];
    }
}

方法二：优先队列

遍历二维数组，把每个元素放入优先队列中(大顶堆)。保证队列的长度不超过k，如果队列的长度大于k则删掉队头元素。遍历结束后队首元素就是所求值。

Java中通过比较器Comparator实现PriorityQueue

示例：新建比较器并重写compare方法

import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Queue;

public class test {
	
	static Comparator<Integer> cmp = new Comparator<Integer>() {
	      public int compare(Integer e1, Integer e2) {
	        return e2 - e1;//若返回正数，即o2>o1的情况下，则需要交换o1，o2的位置，即降序排列desc
	      }
	    };

	public static void main(String[] args) {
		//不用比较器，默认升序排列
        Queue<Integer> q = new PriorityQueue<>();
        q.add(3);
        q.add(2);
        q.add(4);
        while(!q.isEmpty())
        {
            System.out.print(q.poll()+" ");
        }
        /**
         * 输出结果
         * 2 3 4 
         */
        //使用自定义比较器，降序排列
        System.out.println("<<<");
        Queue<Integer> qq = new PriorityQueue<>(cmp);
        qq.add(3);
        qq.add(2);
        qq.add(4);
        while(!qq.isEmpty())
        {
            System.out.print(qq.poll()+" ");
        }
        /**
         * 输出结果
         * 4 3 2 
         */
	}
}

Java的Comparator升序降序的记法

可参考：https://blog.csdn.net/weixin_43691723/article/details/108030395

实现Comparator接口，必须实现下面这个函数：

@Override
public int compare(CommentVo o1, CommentVo o2) {
           return o1.getTime().compareTo(o2.getTime());
}

这里o1表示位于前面的对象，o2表示后面的对象

• 返回-1（或负数），表示不需要交换01和02的位置，o1排在o2前面，asc（PriorityQueue是小根堆，o1<o2）
• 返回1（或正数），表示需要交换01和02的位置，o1排在o2后面，desc

PriorityQueue的常见方法

peek()//返回队首元素
poll()//返回队首元素，队首元素出队列
add()//添加元素
size()//返回队列元素个数
isEmpty()//判断队列是否为空，为空返回true,不空返回false

PriorityQueue 的添加方法有 2 种，分别是add(E e)和offer(E e)，两者语义相同，都是向优先队列中插入元素，只是Queue接口规定二者对插入失败时的处理不同，前者在插入失败时抛出异常，后则返回false

class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        //通过比较器comparator实现PriorityQueue
        PriorityQueue<Integer> pq = new PriorityQueue<>(cmp);//降序排列
        int rows = matrix.length;
        int cols = matrix[0].length;
        for(int i = 0; i < rows; i++){
            for(int j = 0; j < cols; j++){
                pq.offer(matrix[i][j]);
                if(pq.size()>k){
                    pq.poll();
                }
            }
        }
        return pq.peek();//返回队首
    }
    
    Comparator<Integer> cmp = new Comparator<Integer>() {
	    public int compare(Integer e1, Integer e2) {
	        return e2 - e1;//若返回正数，即o2>o1的情况下，则需要交换o1，o2的位置，即降序排列desc
	    }
	};

}