0
点赞
收藏
分享

微信扫一扫

算法力扣刷题记录 六十八【131.分割回文串】

耳一文 2024-08-09 阅读 6

力扣链接:. - 力扣(LeetCode)

思路:用优先队列的直接按出现频率排序

时间复杂度:O(Nlogk),其中 N 为数组的长度。我们首先遍历原数组,并使用哈希表记录出现次数,每个元素需要 O(1) 的时间,共需 O(N) 的时间。随后,我们遍历「出现次数数组」,由于堆的大小至多为 k,因此每次堆操作需要 O(logk) 的时间,共需 O(Nlogk) 的时间。二者之和为 O(Nlogk)

空间复杂度:O(N)。哈希表的大小为 O(N),而堆的大小为 O(k),共计为 O(N)

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> mp = new HashMap<>();
        for (int i : nums) {
            int num = mp.getOrDefault(i, 0) + 1;
            mp.put(i, num);
        }
        PriorityQueue<Integer> que = new PriorityQueue<>((s1, s2) -> mp.get(s2) - mp.get(s1));
        for (int num : mp.keySet()) {
            que.add(num);
        }
        int[] ans = new int[k];
        for (int i = 0; i < k; i++) {
            ans[i] = que.poll();
        }
        return ans;

    }
}
举报

相关推荐

0 条评论