LeetCode 347. Top K Frequent Elements

 

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

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

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

 1 public class Solution {
 2     public List<Integer> topKFrequent(int[] nums, int k) {
 3         Map<Integer, Integer> map = new HashMap<>();
 4         Arrays.stream(nums).forEach(x -> map.merge (x, 1, Integer::sum));
 5         Queue<Integer> queue = new PriorityQueue<>((x, y) -> map.get(x).compareTo(map.get(y)));
 6         map.forEach((key,value) ->{
 7             if (queue.size()<k) {
 8                 queue.add(key);
 9             }
10             else if (map.get(queue.peek())<value){
11                 queue.poll();
12                 queue.add(key);
13             }
14         });
15         List<Integer> list = new ArrayList<>();
16         queue.forEach(list::add);
17         return list;
18     }
19 }

 

 

posted @ 2016-06-28 03:38  活在夢裡  阅读(193)  评论(0编辑  收藏  举报