四数相加 II

/*
 * 思路：
 * 1. 使用 Java Stream API 计算 nums1 和 nums2 的两两和，并存入 Map 中。
 * 2. 使用 Java Stream API 遍历 nums3 和 nums4，并计算其两两和的相反数在 Map 中的出现次数。
 * 3. 最后返回结果。
 */
import java.util.HashMap;
import java.util.Map;
import java.util.Arrays;
 
public class FourSumCountStream {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer, Integer> map = new HashMap<>();
        Arrays.stream(nums1).forEach(i -> 
            Arrays.stream(nums2).forEach(j -> 
                map.put(i + j, map.getOrDefault(i + j, 0) + 1)
            )
        );
 
        return Arrays.stream(nums3).flatMap(k -> 
            Arrays.stream(nums4).mapToObj(l -> -(k + l))
        ).mapToInt(sum -> map.getOrDefault(sum, 0)).sum();
    }
}