【中等】2240. 买钢笔和铅笔的方案数

原题链接：https://leetcode.cn/problems/number-of-ways-to-buy-pens-and-pencils/

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

给你一个整数 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 支铅笔。

c++

class Solution {
public:
    long long waysToBuyPensPencils(int total, int cost1, int cost2) {
        long spent = 0, ret = 0;
        while (spent <= total) {
            ret += 1 + int((total - spent)/cost2);
            spent += cost1;
        }
        return ret;
    }
};

golang

func waysToBuyPensPencils(total int, cost1 int, cost2 int) int64 {
    var spent int
    var ret int64
    for spent <= total {
        ret = ret + 1 + int64((total - spent)/cost2)
        spent += cost1
    }
    return ret
}