Leetcode刷题：LCP 18. 早餐组合

最新推荐文章于 2024-10-08 10:36:00 发布

初♚心i

最新推荐文章于 2024-10-08 10:36:00 发布

阅读量118

点赞数

文章标签： leetcode 算法职场和发展 python

本文链接：https://blog.csdn.net/qq_51560637/article/details/132908280

版权

本文介绍了LeetCode题目的解法，针对早餐组合问题，提出一种优化算法，利用排序和二分查找减少循环次数，以避免时间超限。通过计算staple中每种主食对应饮料的选择数量，求出总方案数并以1e9+7取模。

摘要由CSDN通过智能技术生成

Leetcode刷题：LCP 18. 早餐组合

题目描述

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

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

示例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。

示例2

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

解题思路

这个道题的难点在于时间超限，后面大部分测试样例都会使时间超限，所以暴力解法（双循环），不可行。
如何只循环一次就可以满足要求呢，我们可以知道staple[i]+drinks[j]<=x,那么我们可以遍历staple,用x-staple[i],那么在drinks中找出所有小于x-staple[i]的值就可以啦。
此处补充一个可以直接用的函数:

函数	作用	举例
bisect(arr,num)	与bisect_right相同
bisect_right(arr,num)	将num按照大小插入到arr中，比num小的右边，包括相等的情况，返回下标	arr=[1,2,2,3],num=2 ,那么返回3，也就是2和3的中间
bisect_left(arr,num)	将num按照大小插入到arr中，比num小的左边，不包括相等的情况，返回下标	arr=[1,2,2,3],num=2 ,那么返回1，也就是1和2的中间

注意：arr必须为升序列
返回的下标也就是小于某值的个数。

完整代码

def breakfastNumber(self, staple: List[int], drinks: List[int], x: int) -> int:
        staple.sort()
        drinks.sort()
        step=0
        for i in staple:
            b=bisect_right(drinks,x-i)
            step+=b

        return step%(10**9+7)