手撸早餐组合问题解法

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

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

示例:输入：staple = [10,20,5], drinks = [5,5,2], x = 15    输出：6

解释：小扣有 6 种购买方案，所选主食与所选饮料在数组中对应的下标分别是：
第 1 种方案：staple[0] + drinks[0] = 10 + 5 = 15；
第 2 种方案：staple[0] + drinks[1] = 10 + 5 = 15；
第 3 种方案：staple[0] + drinks[2] = 10 + 2 = 12；
第 4 种方案：staple[2] + drinks[0] = 5 + 5 = 10；
第 5 种方案：staple[2] + drinks[1] = 5 + 5 = 10；
第 6 种方案：staple[2] + drinks[2] = 5 + 2 = 7。

 

排序加双指针解法

class Solution {
    public int breakfastNumber(int[] staple, int[] drinks, int x) {
        Arrays.sort(staple);
        Arrays.sort(drinks);
        int count = 0;
        for (int i = 0, j = drinks.length - 1; i < staple.length && j >= 0;) {
            if (staple[i] + drinks[j] > x) {
                j--;
            } else {
                count += j + 1;
                count = count % 1000000007;
                i++;
            }
        }
        return count;
    }
}

1.将staple，drinks从小到大排序
2.设置变量i(i从0到staple.length-1),j(j从drinks.length-1到0)
让staple[i]+drinks[j]与x比较
一开始staple[0]+drinks[drinks.length-1] > x,说明花费已经超过x了，需要减少饮料的价格，故j--
一开始staple[0]+drinks[drinks.length-1] <= x,说明花费符合要求，由于变量j是从大到小的，则说明前面的drinks前面的元素也必然符合要求，没必要再去多此一举判断一遍，
直接让计数累加上j+1(j+1表示符合要求的饮料价格有几个)就完事了