题目：
给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
提示：
- 你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
- 你的算法的时间复杂度必须优于 O(n log n) ,n 是数组的大小。
- 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
- 你可以按任意顺序返回答案。
解法1：heap
/**
* 思路：
* 统计词频在新的map上，k是数，v是词频
* heap中存放词频，统计前k个
*/
public int[] topKFrequent(int[] nums, int k) {
int[] res=new int[k];
HashMap<Integer, Integer> map = new HashMap<>();
PriorityQueue<Integer> heap = new PriorityQueue<>((v1, v2) -> map.get(v2) - map.get(v1));
for (int i:nums){
map.put(i,map.getOrDefault(i, 0)+1);
}
for(int key:map.keySet()){
heap.add(key);
}
for (int i=0;i<k;i++){
res[i]=heap.remove();
}
return res;
}
时间复杂度：Onlogn
空间复杂度：On
解法2：二叉搜索树
/**
* 思路：
* 二叉搜索树
* 统计词频在新的map上，k是数，v是词频
* TreeMap中存放，k是词频，v是数的集合
* 之后获取前k高词频的数
*/
public int[] topKFrequent(int[] nums, int k) {
int[] result = new int[k];
HashMap<Integer, Integer> map = new HashMap<>();
for(int i:nums){
map.put(i,map.getOrDefault(i,0)+1);
}
TreeMap<Integer, List<Integer>> bST = new TreeMap<>();
for (int num:map.keySet()){
Integer frequent = map.get(num);
if (!bST.containsKey(frequent)){
bST.put(frequent,new ArrayList());
}
bST.get(frequent).add(num);
}
List<Integer> list = new ArrayList<>();
while (list.size()<k){
list.addAll(bST.pollLastEntry().getValue());
}
for (int i=0;i<k;i++){
result[i]= list.get(i);
}
return result;
}
时间复杂度：On
空间复杂度：On
