力扣LeetCode刷题心得之Python 买钢笔和铅笔的方案数

6061. 买钢笔和铅笔的方案数

给你一个整数 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 <= 106

来源：力扣

解题思路：

第一步，光买钢笔有几种方案

 a = int(total/cost1)+1

因为0只钢笔也算一种方案，所以加一

第二步，买钢笔后，剩下的钱能购买多少铅笔

int((total-cost1*n)/cost2)

第三步，在买钢笔的几种方案下，剩下的钱购买铅笔有几种方案

for n in range(a):
    int((total-cost1*n)/cost2)+1

第四步，将所有方案相加

b = 0
for n in range(a):
    b += int((total-cost1*n)/cost2)+1

代码整合：

class Solution(object):
    def waysToBuyPensPencils(self, total, cost1, cost2):
        """
        :type total: int
        :type cost1: int
        :type cost2: int
        :rtype: int
        """
        a = int(total/cost1)+1
        b = 0
        for n in range(a):
            b += int((total-cost1*n)/cost2)+1
        return b