Java——PriorityQueue使用forEach输出元素乱序并实现逆序输出

文章目录

  • 问题发现
  • 原因
  • 正确输出方式
    • 正序输出
    • 倒序输出
      • 方法一:自定义比较器
      • 方法二:使用负数

问题发现

代码如下:

public class Main {

    public static void main(String[] args) {
        PriorityQueue<Double> pq = new PriorityQueue<>();
        pq.add(3.6);
        pq.add(1.2);
        pq.add(5.9);
        pq.add(2.4);
        pq.forEach(System.out::println);
    }
}

输出结果:

1.2
2.4
5.9
3.6

可以看到输出的顺序既不是添加顺序也不是按照元素大小顺序输出

原因

PriorityQueue 使用 forEach 方法输出元素乱序的原因与其内部实现有关。forEach 方法是对集合进行迭代操作,它并不保证按照特定的顺序遍历元素。在 PriorityQueue 的实现中,元素的顺序是根据堆结构决定的,而不是插入的顺序或者元素自身的顺序。

PriorityQueue 是使用堆实现的,堆是一种完全二叉树的数据结构,它具有以下特性:

  1. 对于最小堆,父节点的值小于等于子节点的值。
  2. 对于最大堆,父节点的值大于等于子节点的值。

当你向 PriorityQueue 中添加元素时,它会根据优先级规则(最小堆或最大堆)进行调整,以保持堆的特性。因此,堆中的元素的顺序并不是按照插入的顺序或者元素自身的顺序。

而使用 PriorityQueueforEach 方法对堆进行迭代时,它会按照内部的二叉树结构进行遍历,而不是按照元素的实际顺序。这就导致了在使用 forEach 方法输出元素时,它们的顺序是乱序的。

如果希望按照特定顺序遍历 PriorityQueue 中的元素,你可以使用其他方式,如循环遍历或使用迭代器来逐个取出元素。

正确输出方式

正序输出

代码如下:

public class Main {

    public static void main(String[] args) {
        PriorityQueue<Double> pq = new PriorityQueue<>();
        pq.add(3.6);
        pq.add(1.2);
        pq.add(5.9);
        pq.add(2.4);
//        pq.forEach(System.out::println);
        while (!pq.isEmpty()){
            System.out.println(pq.poll());
        }
    }
}

输出结果:

1.2
2.4
3.6
5.9

可以发现输出的顺序是按照元素大小正序输出的

倒序输出

方法一:自定义比较器

你可以创建一个自定义的比较器类,然后在创建 PriorityQueue 对象时将该比较器作为参数传入。比较器将比较两个元素的值,并根据需要对它们的顺序进行调整。

示例代码:

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

public class Main {

    public static void main(String[] args) {
        // 创建自定义的比较器
        Comparator<Double> reverseComparator = new Comparator<Double>() {
            public int compare(Double d1, Double d2) {
                return Double.compare(d2, d1); // 逆序比较
            }
        };

        // 创建带有自定义比较器的 PriorityQueue
        PriorityQueue<Double> pq = new PriorityQueue<>(reverseComparator);

        // 添加元素
        pq.add(3.6);
        pq.add(1.2);
        pq.add(5.9);
        pq.add(2.4);

        // 输出元素(按逆序顺序)
        while (!pq.isEmpty()) {
            System.out.println(pq.poll());
        }
    }
}

输出结果:

5.9
3.6
2.4
1.2

方法二:使用负数

另一种简单的方法是,将要存储的 Double 值取其负数,然后正常地使用 PriorityQueue。这样一来,元素将以逆序的方式进行排序。

示例代码:

import java.util.PriorityQueue;

public class Main {

    public static void main(String[] args) {
        // 创建 PriorityQueue
        PriorityQueue<Double> pq = new PriorityQueue<>();

        // 添加元素(取负数)
        pq.add(-3.6);
        pq.add(-1.2);
        pq.add(-5.9);
        pq.add(-2.4);

        // 输出元素(按逆序顺序,注意取负数后要再次取负数)
        while (!pq.isEmpty()) {
            System.out.println(-pq.poll());
        }
    }
}

输出结果:

5.9
3.6
2.4
1.2

你可能感兴趣的:(Java学习,Java,数据结构,优先级队列,PriorityQueue)