前 K 个高频元素

/*
 * 思路：
 * 1. 使用Stream和Collectors来统计每个数字的频率。
 * 2. 使用Stream来获取频率最高的k个元素。
 * 3. 使用Stream和Collectors将结果转换为数组。
 */
import java.util.*;
import java.util.stream.Collectors;

public class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        // 使用Stream统计频率
        Map<Integer, Long> frequencyMap = Arrays.stream(nums)
            .boxed()
            .collect(Collectors.groupingBy(e -> e, Collectors.counting()));
        // 使用Stream获取前k个频率最高的元素
        List<Integer> topK = frequencyMap.entrySet().stream()
            .sorted((a, b) -> Long.compare(b.getValue(), a.getValue()))
            .limit(k)
            .map(Map.Entry::getKey)
            .collect(Collectors.toList());
        // 转换为数组并返回
        return topK.stream().mapToInt(Integer::intValue).toArray();
    }
}