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

6060. 找到最接近 0 的数字

给你一个长度为 n 的整数数组 nums ，请你返回 nums 中最 接近 0 的数字。如果有多个答案，请你返回它们中的 最大值 。

示例 1：

输入：nums = [-4,-2,1,4,8]
输出：1
解释：
-4 到 0 的距离为 |-4| = 4 。
-2 到 0 的距离为 |-2| = 2 。 1 到 0 的距离为 |1| = 1 。 4 到 0 的距离为 |4| = 4 。 8 到 0 的距离为 |8| = 8 。 所以，数组中距离 0 最近的数字为 1 。

示例 2：

输入：nums = [2,-1,1]
输出：1
解释：1 和 -1 都是距离 0 最近的数字，所以返回较大值 1 。

提示：

1 <= n <= 1000
-105 <= nums[i] <= 105

解题思路：

第一步，求出每个数到0的距离

for i in nums:
    n.append(abs(i-0))

第二步，最小距离的索引值

m = [x for x ,y in list(enumerate(n)) if y ==min(n)]

第三步，距离0最近的数字

for k in m:
    p.append(nums[k])

第四步，返回较大值

return max(p)

代码整合：

class Solution(object):
    def findClosestNumber(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n = []
        p = []
        m = []
        for i in nums:
            n.append(abs(i-0))
        m = [x for x ,y in list(enumerate(n)) if y ==min(n)]
        for k in m:
            p.append(nums[k])
        return max(p)

优化后：

class Solution(object):
    def findClosestNumber(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        nums.sort(key = lambda x : (abs(x), -x))
        return nums[0]