字节跳动青训营——入营考核解答（持续更新中~~~）

AI 加码，字节跳动青训营，等待您的加入~

1、报名方式

1. 点击以下链接：字节跳动青训营报名入口
2. 扫描图片二维码：

2、考核内容

在指定的题库中自主选择不少于 15 道算法题并完成解题，其中题目难度分配如下：

• 简单题不少于 10 道
• 中等题不少于 4 道
• 困难题不少于 1 道

解答代码

27. 查找热点数据（困难）

代码实现：

import java.util.*;

public class Main {
    public static List<Integer> solution(int[] nums, int k) {
        // 首先创建一个 Map 来存储每个数字及其出现的次数
        Map<Integer, Integer> frequencyMap = new HashMap<>();
        for (int num : nums) {
            if (frequencyMap.containsKey(num)) {
                // 如果数字已经在 Map 中，增加其出现次数
                frequencyMap.put(num, frequencyMap.get(num) + 1);
            } else {
                // 如果数字不在 Map 中，将其添加并初始化为 1
                frequencyMap.put(num, 1);
            }
        }

        // 创建一个优先级队列来存储频率最高的元素
        PriorityQueue<Map.Entry<Integer, Integer>> maxHeap = new PriorityQueue<>(
                (entry1, entry2) -> entry2.getValue() - entry1.getValue());

        // 将 Map 中的元素添加到优先级队列中
        for (Map.Entry<Integer, Integer> entry : frequencyMap.entrySet()) {
            maxHeap.offer(entry);
        }

        // 取出前 k 个高频元素
        List<Integer> result = new ArrayList<>();
        for (int i = 0; i < k; i++) {
            result.add(maxHeap.poll().getKey());
        }

        return result;
    }

    public static void main(String[] args) {
        // You can add more test cases here
        int[] nums1 = { 1, 1, 1, 2, 2, 3 };
        int[] nums2 = { 1 };

        System.out.println(solution(nums1, 2).equals(Arrays.asList(1, 2)));
        System.out.println(solution(nums2, 1).equals(Arrays.asList(1)));
    }
}

运行结果：