咒语和药水的成功对数

/*
题目思路：
1. 使用 Java Stream 遍历 spells 数组中的每个元素。
2. 对于每个元素，过滤出 potions 中与之组合后能量强度大于等于 success 的药水数量。
3. 将这些数量存储在结果数组 pairs 中。
*/

import java.util.Arrays;

public class SuccessfulPairsStream {
    public int[] successfulPairs(int[] spells, int[] potions, long success) {
        return Arrays.stream(spells)
                     .map(spell -> (int) Arrays.stream(potions)
                                              .filter(potion -> (long) spell * potion >= success)
                                              .count())
                     .toArray();
    }
}