1. Implementação da fila prioritária
Na implementação da fila de prioridade, podemos escolher a estrutura de dados do heap, a maior fila de prioridade pode usar um grande heap e a menor fila de prioridade pode ser implementada com um pequeno heap. O heap é uma árvore binária, conforme mostrado na figura abaixo de. O princípio é explicado abaixo com a fila de prioridade menor. A fila de prioridade mínima geralmente inclui inserir um elemento no conjunto S, retornar o elemento com a menor chave no conjunto S, retornar e deletar o elemento com a menor chave no conjunto S e assim por diante.
Operação de inserção
A operação de inserção consiste em inserir um elemento no conjunto S, primeiro colocar o elemento na próxima posição de todos os elementos e, em seguida, realizar a operação "flutuar", conforme mostrado no exemplo abaixo (observe que o exemplo na figura abaixo é uma pilha pequena e a imagem é de um entendimento profundo do Java PriorityQueue)
Remover operação
Na fila de prioridade, quando a fila não está vazia, remova o primeiro elemento do conjunto, ou seja, o elemento com subscrito 0 e, em seguida, mova o último elemento do conjunto para a posição com subscrito 0. O novo elemento executa um operação de "afundamento". O exemplo de figura a seguir (observe, mas o princípio é o mesmo, a imagem vem de um entendimento profundo de Java PriorityQueue)
Aprendizagem e uso ao vivo: um
problema de algoritmo:
题目描述:
对于一个数组,寻找其前K个最小的元素
import java.util.Comparator;
import java.util.PriorityQueue;
class Solution {
public void getLeastNumbers(int[] arr, int k) {
int[] num = new int[k];
//Queue<Integer> queue = new PriorityQueue<>((v1, v2) -> v2 - v1);
PriorityQueue<Integer> queue = new PriorityQueue<>(k, new Comparator<Integer>() {
@Override
public int compare(Integer t1, Integer t2) {
return t2 - t1;
}//大根堆
});
for (int value : arr) {
if (queue.size() < k)
queue.offer(value);
else {
if (value < queue.peek()) {
queue.poll();
queue.offer(value);
}
}
}
for (int i = 0; i < k; i ++){
num[i] = queue.poll();
}
for(int i=0;i<k;i++)
System.out.print(num[i]+" ");
//return num;
}
public static void main(String[] args) {
Solution s=new Solution();
int []arr={
4,5,1,6,2,7,3,8};
int k=4;
s.getLeastNumbers(arr,4);
}
}
返回结果:4 3 2 1