LCP 18. 早餐组合

leetcode力扣刷题打卡

题目：LCP 18. 早餐组合
描述：小扣在秋日市集选择了一家早餐摊位，一维整型数组 staple 中记录了每种主食的价格，一维整型数组 drinks 中记录了每种饮料的价格。小扣的计划选择一份主食和一款饮料，且花费不超过 x 元。请返回小扣共有多少种购买方案。

注意：答案需要以 1e9 + 7 (1000000007) 为底取模，如：计算初始结果为：1000000008，请返回 1

解题思路

1、先排序；
2、双循环；固定主食，选择饮料的边界j，此时的主食就有j + 1种配饮料方法；
3、这种双循环都要注意第二个循环参数的边界，比如j >= 0;

原代码##

class Solution {
public:
    int breakfastNumber(vector<int>& staple, vector<int>& drinks, int x) {
        int ns = staple.size(), nd = drinks.size();
        sort(staple.begin(), staple.end());
        sort(drinks.begin(), drinks.end());
        int j = nd - 1;
        long long ans = 0;
        for (int i = 0; i < ns; ++i) {
            if (staple[i] > x) break;
            while (j >= 0 && staple[i] + drinks[j] > x) {
                j--;
            }
            ans += j + 1;
        }
        return ans % 1000000007;
    }
};