算法初探LeetCode-买钢笔和铅笔的方案数

169 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 27 天,点击查看活动详情

LeetCode2240:买钢笔和铅笔的方案数

给你一个整数 total ,表示你拥有的总钱数。同时给你两个整数 cost1 和 cost2 ,分别表示一支钢笔和一支铅笔的价格。你可以花费你部分或者全部的钱,去买任意数目的两种笔。

请你返回购买钢笔和铅笔的 不同方案数目 。

示例 1:

输入: total = 20, cost1 = 10, cost2 = 5
输出: 9
解释: 一支钢笔的价格为 10 ,一支铅笔的价格为 5 。
- 如果你买 0 支钢笔,那么你可以买 0 ,1 ,2 ,3 或者 4 支铅笔。
- 如果你买 1 支钢笔,那么你可以买 0 ,1 或者 2 支铅笔。
- 如果你买 2 支钢笔,那么你没法买任何铅笔。
所以买钢笔和铅笔的总方案数为 5 + 3 + 1 = 9 种。

示例 2:

输入: total = 5, cost1 = 10, cost2 = 10
输出: 1
解释: 钢笔和铅笔的价格都为 10 ,都比拥有的钱数多,所以你没法购买任何文具。所以只有 1 种方案:买 0 支钢笔和 0 支铅笔。

提示:

  • 1<=total,cost1,cost2<=1061 <= total, cost1, cost2 <= 10^6

思路分析

根据题意可知总钱数/钢笔价格得到可以买多少钢笔,枚举钢笔个数时候最大的铅笔数。

如: 20 10 5 最多买俩钢笔 当不买钢笔时,最多4个铅笔。0,1,2,3,4,5 5种方法。

可以推断: 每个钢笔的情况的时候的最大铅笔数+1就是选择个数,累加起来就行。

算法代码

public long waysToBuyPensPencils(int total, int cost1, int cost2) {
    int pen = total / cost1;
    long totalNum = 0;
    for (int i = 0; i < pen + 1; i++) {
        totalNum += (total - cost1 * i) / cost2 + 1;
    }
    return totalNum;
}

结果详情

Snipaste_2023-03-02_22-52-00.png

算法复杂度

  • 空间复杂度:O(1)O(1)
  • 时间复杂度:O(n)O(n)

掘金(JUEJIN)一起进步,一起成长!