LeetCode 347. 前 K 个高频元素

最新推荐文章于 2024-10-16 10:13:45 发布

一杯茶一道题

最新推荐文章于 2024-10-16 10:13:45 发布

阅读量146

点赞数 2

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_64381458/article/details/139560142

版权

LeetCode 专栏收录该内容

55 篇文章 1 订阅

订阅专栏

更多题解尽在 https://sugar.matrixlab.dev/algorithm 每日更新。
组队打卡，更多解法等你一起来参与哦！

LeetCode 347. 前 K 个高频元素，难度中等。

暴力

解题思路：使用 Map 统计每个数字的频率，然后根据 value 大小排序。

import java.util.*;

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer, Integer> mp = new HashMap<>();
        for (int num : nums) {
            if (mp.containsKey(num)) {
                mp.put(num, mp.get(num) + 1);
            } else {
                mp.put(num, 1);
            }
        }
        List<Map.Entry<Integer, Integer>> lst = new ArrayList<>(mp.entrySet());
        lst.sort(Map.Entry.comparingByValue());
        int[] result = new int[k];
        for (int i = 0; i < k; ++i) {
            result[i] = lst.get(lst.size() - i - 1).getKey();
        }
        return result;
    }
}