题目:
给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
我自己的解题思路是用哈希表求频率,然后再通过排序来解决问题,
class Solution {
public List<Integer> topKFrequent(int[] nums, int k) {
Map map = new HashMap<Integer,Integer >();
List<Integer> ll = new ArrayList<Integer>();
for(int i = 0;i<nums.length;i++) {
if(map.containsKey(nums[i])) {
int max = (Integer)map.get(nums[i])+1;
map.put(nums[i],max);
}else {
map.put(nums[i], 1);
}
}
List<Map.Entry<Integer, Integer>> list = new ArrayList<Map.Entry<Integer,Integer>>();
list.addAll(map.entrySet());
System.out.println(list);
Collections.sort(list, new Comper());
for(int i=0;i<list.size();i++) {
System.out.println(i);
int value = list.get(i).getKey();
ll.add(value);
}
return ll;
}
}
class Comper implements Comparator<Map.Entry<Integer,Integer>>{
public int compare(Entry<Integer, Integer> o1, Entry<Integer, Integer> o2) {
return o1.getValue() - o2.getValue();
}
}
这个解题思路很清新,别具一格,膜拜大佬的感觉
前面求取频率都是一样的做法,亮点在后面,通过下标来排序,这里也有一个缺点,就是没有考虑次数一样多的情况,但是这个代码很厉害
class Solution {
public List<Integer> topKFrequent(int[] nums, int k) {
List<Integer> res = new ArrayList();
// key为值 value为出现次数
HashMap<Integer,Integer> map = new HashMap();
// 统计每个数出现的次数
for(int i : nums){
map.put(i,map.getOrDefault(i,0)+1);
}
// list数组中每一个位置为这个数出现的次数 值为出现次数这么多的数
List<Integer>[] list = new List[nums.length+1];
for(int key : map.keySet()){
// 出现的次数
int i = map.get(key);
if(list[i]==null) list[i] = new ArrayList();
// 当前次数中 包含那些数
list[i].add(key);
}
for(int i=list.length-1;i>=0 && res.size()<k;i--){
if(list[i]==null) continue;
// 该题没有出现频率相同的数字 不需要考虑其他情况
res.addAll(list[i]);
}
return res;
}
}
练习题
最新推荐文章于 2024-03-20 18:58:53 发布