开启掘金成长之旅!这是我参与「掘金日新计划 · 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 支铅笔。
提示:
思路分析
根据题意可知总钱数/钢笔价格得到可以买多少钢笔,枚举钢笔个数时候最大的铅笔数。
如: 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;
}
结果详情
算法复杂度
- 空间复杂度:
- 时间复杂度:
在掘金(JUEJIN)一起进步,一起成长!