347. 前 K 个高频元素 难度中等
给定一个非空的整数数组,返回其中出现频率前 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 是数组的大小。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
执行用时 :10 ms, 在所有 Java 提交中击败了99.43% 的用户
内存消耗 :42.3 MB, 在所有 Java 提交中击败了5.25%的用户
class Solution {
public List<Integer> topKFrequent(int[] nums, int k) {
Arrays.sort(nums);
HashMap<Integer, Integer> hashMap = new HashMap<>();
hashMap.put(nums[0], 1);
int temp = 1;
int i;
for (i = 1; i < nums.length; ++i) {
if (nums[i] == nums[i - 1]) {
temp++;
} else {
hashMap.put(nums[i - 1], temp);
temp = 1;
}
}
hashMap.put(nums[nums.length - 1], temp);
List<Map.Entry<Integer, Integer>> listMap = new ArrayList<>(hashMap.entrySet());
listMap.sort(new Comparator<Map.Entry<Integer, Integer>>() {
@Override
public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
return o2.getValue().compareTo(o1.getValue());
}
});
List<Integer> ans = new ArrayList<>();
for (int j = 0; j < k; ++j) {
ans.add(listMap.get(j).getKey());
}
return ans;
}
}