早餐组合

/*
 * 思路：
 * 1. 对主食数组 staple 进行排序。
 * 2. 使用流操作遍历饮料数组 drinks，对于每一个饮料价格，使用二分查找在主食数组中找到可以搭配的最大主食价格。
 * 3. 计算满足条件的主食数量并累加到总数中。
 * 4. 最后对结果取模 1e9 + 7。
 */
import java.util.Arrays;
import java.util.stream.IntStream;

public class BreakfastComboStream {
    public int breakfastNumber(int[] staple, int[] drinks, int x) {
        Arrays.sort(staple);
        int MOD = 1000000007;
        return IntStream.of(drinks).map(drink -> {
            int maxStaple = x - drink;
            if (maxStaple <= 0) return 0;
            int left = 0, right = staple.length - 1;
            while (left <= right) {
                int mid = left + (right - left) / 2;
                if (staple[mid] <= maxStaple) {
                    left = mid + 1;
                } else {
                    right = mid - 1;
                }
            }
            return left;
        }).reduce(0, (a, b) -> (a + b) % MOD);
    }
}