Given a non-empty array of integers, return the k most frequent elements.

Example 1:

Input: nums = [1,1,1,2,2,3], k = 2
Output: [1,2]

Example 2:

Input: nums = [1], k = 1
Output: [1]

---

这个题目更加像是一道语法题目。。。

首先得知道map该如何遍历

方式一 这是最常见的并且在大多数情况下也是最可取的遍历方式。在键值都需要时使用。

1 2 3 4

Map<Integer, Integer> map = new HashMap<Integer, Integer>(); for (Map.Entry<Integer, Integer> entry : map.entrySet()) {   System.out.println("Key = " + entry.getKey() + ", Value = " + entry.getValue()); }

方法二 在for-each循环中遍历keys或values。

如果只需要map中的键或者值，你可以通过keySet或values来实现遍历，而不是用entrySet。

1 2 3 4 5 6 7 8 9

Map<Integer, Integer> map = new HashMap<Integer, Integer>(); //遍历map中的键 for (Integer key : map.keySet()) {   System.out.println("Key = " + key); } //遍历map中的值 for (Integer value : map.values()) {   System.out.println("Value = " + value); }

在Java中map的键值对Map.Entry对应c++中的pair。

public List<Integer> topKFrequent(int[] nums, int k) {
        List<Integer> list = new ArrayList<>();

        PriorityQueue<Map.Entry<Integer, Integer>> pq = new PriorityQueue<>(new Comparator<Map.Entry<Integer, Integer>>() {
            @Override
            public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
                return o2.getValue() - o1.getValue();
            }
        });

        Map<Integer, Integer> map = new HashMap<>();

        for (int i = 0; i < nums.length; i++)
            if (map.containsKey(nums[i]))
                map.put(nums[i], map.get(nums[i]) + 1);
            else
                map.put(nums[i], 1);
//        多此一举题目中不可能出现相同的键值对
//        Set<Map.Entry<Integer, Integer>> set = map.entrySet();
//
//        for (Map.Entry<Integer, Integer> entry : set)
//            pq.add(entry);
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            pq.add(entry);
        }
        for (int i = 0; i < k; i++)
            list.add(pq.poll().getKey());

        return list;
    }