基于PriorityQueue的HashMap自定义排序问题排查与结果修正

核心问题1：PriorityQueue的打印误区 #

PriorityQueue的toString()方法直接输出内部存储的数组元素，不保证按优先级顺序排列。它基于小顶堆实现，只有通过poll()方法逐个取出元素时，才会按照设定的优先级顺序返回结果。你直接打印pq得到的是堆的内部结构，并非排序后的最终结果。

核心问题2：潜在的空指针风险 #

如果Map中存在null键或null值，map.get(a).equals(map.get(b))会抛出NullPointerException——因为map.get(a)可能返回null，调用equals()方法会触发空指针异常。

修复后的代码 #

import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;

public class Main {
    public static void main(String[] args) {

        Map<String, Integer> map = new HashMap<>() {{
            put("Hello", 50);
            put("Yes", 50);
            put("No", 100);
        }};
        PriorityQueue<String> pq = new PriorityQueue<>((a, b) -> {
            Integer valA = map.get(a);
            Integer valB = map.get(b);
            // 若业务场景存在null值，可补充null判断逻辑，此处假设值非null
            if (valA.equals(valB)) {
                return a.compareTo(b); // 值相等时按键升序排列
            } else {
                return Integer.compare(valB, valA); // 值降序排列
            }
        });
        verifyPQ(map, pq);
        
        // 正确获取排序结果：通过poll()逐个取出元素
        System.out.print("PQ: [");
        boolean first = true;
        while (!pq.isEmpty()) {
            if (!first) {
                System.out.print(", ");
            }
            System.out.print(pq.poll());
            first = false;
        }
        System.out.println("]");
    }

    private static void verifyPQ(Map<String, Integer> map, PriorityQueue<String> pq) {
        for (String key : map.keySet()) {
            pq.add(key);
            if (pq.size() > 2) {
                pq.poll();
            }
        }
    }
}

修复说明 #

1. 正确获取排序结果：替换直接打印pq的方式，通过poll()循环取出元素，此时得到的是严格按照优先级排序的结果，输出为PQ: [No, Hello]，符合预期。
2. 规避空指针：先将map.get(a)和map.get(b)赋值给局部变量再做比较，避免直接调用equals()触发NPE。若业务场景可能出现null值，可补充额外的null判断逻辑（比如将null值视为最小或最大）。
3. 比较器逻辑验证：原比较器的逻辑是正确的——值相等时按键升序，值不等时按值降序，最终保留前2个优先级最高的元素（值最大的两个，值相同时键较小的优先）。

内容的提问来源于stack exchange，提问作者Shubham Kumbhalkar