Explicação detalhada da fila de prioridade Java

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)

Insira a descrição da imagem aqui
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)

Insira a descrição da imagem aqui
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 

Acho que você gosta

Origin blog.csdn.net/fxwentian/article/details/115195995
Recomendado
Clasificación