LeetCode 347. Top K Frequent Elements

 

题目链接：https://leetcode.com/problems/top-k-frequent-elements/

最近在专门学java 8的特性，这题我主要是用来练一下写java 8的，题目的算法可以写个基于堆的nlog(k)，这个没什么好多说的了。

不保证我的java 8程序是最简洁的，我很可能还是写了啰嗦了。

public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        Arrays.stream(nums).forEach(x -> map.merge (x, 1, Integer::sum));
        Queue<Integer> queue = new PriorityQueue<>((x, y) -> map.get(x).compareTo(map.get(y)));
        map.forEach((key,value) ->{
            if (queue.size()<k) {
                queue.add(key);
            }
            else if (map.get(queue.peek())<value){
                queue.poll();
                queue.add(key);
            }
        });
        List<Integer> list = new ArrayList<>();
        queue.forEach(list::add);
        return list;
    }
}