当前位置: 代码迷 >> 综合 >> 手撸早餐组合问题解法
  详细解决方案

手撸早餐组合问题解法

热度:13   发布时间:2024-02-27 10:02:43.0

小扣在秋日市集选择了一家早餐摊位,一维整型数组 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表示符合要求的饮料价格有几个)就完事了

  相关解决方案