有人能帮我理解优先队列吗？[闭门]

Question

    class Solution {
    public List<String> topKFrequent(String[] words, int k) {
        List<String> answer = new ArrayList();
        HashMap<String, Integer> frequency = new HashMap<>();
        //record the frequency
        for (String word: words){
            frequency.put(word,frequency.getOrDefault(word,0)+1);
        }
        //record the frequency in the priority queue
        PriorityQueue<String> pq = new PriorityQueue<>((a,b)->{
            if(frequency.get(a) != frequency.get(b)){
                return frequency.get(b) - frequency.get(a);
            }else{
                return frequency.get(b).compareTo(frequency.get(a));
            }
        });
        for(String key:frequency.keySet()){
            pq.offer(key);
        }
        //return the first kth string
        for(int i = 0; i < k;i++){
            answer.add(pq.remove());
        }
        return answer;
    }
}

答案应该是[“我”，“爱”]而我得到了[“爱”，“我”]我想因为在pq中，爱在我之前，所以可能我在pq中做错了，但我不知道如何解决这个问题

Answer 1

我认为问题不在于你对优先队列的理解。由于其他原因，此代码失败。

第一：

frequency.get(a) != frequency.get(b)

您正在比较对象，而不是原始int。您可以通过将其中一个对象转换为这样的对象来解决这个问题

(int) frequency.get(a) != frequency.get(b)

第二：如果两者的频率相同，则需要按字母顺序进行比较。此代码：

frequency.get(b).compareTo(frequency.get(a))

再次比较它们的频率。把它换成这个

a.compareTo(b)

这两个改变应该可以解决你的问题。