数组中紧跟 key 之后出现最频繁的数字

// Java Stream Solution
// 题目思路：
// 1. 使用Stream API统计key后一个元素的出现次数
// 2. 通过Collectors.groupingBy和Collectors.counting收集数据
// 3. 找到出现次数最多的元素并返回

import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Solution {
    public int mostFrequentTarget(int[] nums, int key) {
        Map<Integer, Long> countMap = IntStream.range(0, nums.length - 1)
            .filter(i -> nums[i] == key)
            .mapToObj(i -> nums[i + 1])
            .collect(Collectors.groupingBy(t -> t, Collectors.counting()));
        return countMap.entrySet().stream()
            .max(Map.Entry.comparingByValue())
            .get()
            .getKey();
    }
}