【二分查找】LCP 18. 早餐组合

📍前言

🕺作者: 迷茫的启明星

学习路线
C语言从0到1
C++初阶
数据结构从0到1

😘欢迎关注:👍点赞🙌收藏✍️留言

🏇码字不易,你的👍点赞🙌收藏❤️关注对我真的很重要,有问题可在评论区提出,感谢阅读!!!

持续更新中~

【二分查找】LCP 18. 早餐组合

在另一篇博客里讲过二分法的模板:
《二分法的模板讲解》

题目描述:

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

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

解题思路:

首先,我们需要对主食和饮料的价格进行排序。这样,我们可以在遍历主食价格的同时,使用二分查找法在有序的饮料价格中找到合适的饮料价格,使得主食和饮料的总价格不超过 x 元。

为了实现二分查找,我们需要定义左右指针 leftright,以及中间指针 mid。在每一次循环中,我们比较 drinks[mid] 与目标价格 target,即 x - staple[i]。如果 drinks[mid] 小于等于 target,说明饮料价格还有可能在左侧区间,所以我们将 left 指针更新为 mid + 1。否则,我们将 right 指针更新为 mid - 1

当遍历完所有的主食价格后,我们将得到的购买方案数量 count 对 1e9 + 7 取模,最后返回 count

代码实现:

C++ 复制代码
class Solution {
public:
    int breakfastNumber(vector<int>& staple, vector<int>& drinks, int x) {
        sort(staple.begin(), staple.end());
        sort(drinks.begin(), drinks.end());
        int count = 0;
        for (int i = 0; i < staple.size(); ++i) {
            int target = x - staple[i];
            int left = 0, right = drinks.size() - 1;
            while (left <= right) {
                int mid = left + (right - left) / 2;
                if (drinks[mid] <= target) {
                    left = mid + 1;
                } else {
                    right = mid - 1;
                }
            }
            count = (count + left) % ***;
        }
        return count;
    }
};

总结:

这道题目考察了排序和二分查找法在求解组合问题中的应用。通过将主食和饮料的价格排序,我们可以在 O(nlogn) 的时间复杂度内完成购买方案的查找。而二分查找法在这里起到了关键作用,使得我们可以在 O(logn) 的时间复杂度内找到合适的饮料价格。

相关推荐
地平线开发者14 小时前
SparseDrive 模型导出与性能优化实战
算法·自动驾驶
董董灿是个攻城狮14 小时前
大模型连载2:初步认识 tokenizer 的过程
算法
地平线开发者15 小时前
地平线 VP 接口工程实践(一):hbVPRoiResize 接口功能、使用约束与典型问题总结
算法·自动驾驶
罗西的思考15 小时前
AI Agent框架探秘:拆解 OpenHands(10)--- Runtime
人工智能·算法·机器学习
HXhlx18 小时前
CART决策树基本原理
算法·机器学习
Wect19 小时前
LeetCode 210. 课程表 II 题解:Kahn算法+DFS 双解法精讲
前端·算法·typescript
颜酱19 小时前
单调队列:滑动窗口极值问题的最优解(通用模板版)
javascript·后端·算法
肆忆_1 天前
# 用 5 个问题学懂 C++ 虚函数(入门级)
c++
不想写代码的星星1 天前
虚函数表:C++ 多态背后的那个男人
c++
Gorway1 天前
解析残差网络 (ResNet)
算法