1.题目
小力将 N 个零件的报价存于数组 nums
。小力预算为 target
,假定小力仅购买两个零件,要求购买零件的花费不超过预算,请问他有多少种采购方案。
注意:答案需要以 1e9 + 7 (1000000007)
为底取模,如:计算初始结果为:1000000008
,请返回 1
示例 1:
输入:
nums = [2,5,3,5], target = 6
输出:
1
解释:预算内仅能购买 nums[0] 与 nums[2]。
示例 2:
输入:
nums = [2,2,1,9], target = 10
输出:
4
解释:符合预算的采购方案如下: nums[0] + nums[1] = 4 nums[0] + nums[2] = 3 nums[1] + nums[2] = 3 nums[2] + nums[3] = 10
提示:
2 <= nums.length <= 10^5
1 <= nums[i], target <= 10^5
2.解析
这个代码的目的是找出有多少种购买方案可以达到特定的"target"值。在给定的"nums"列表中,每个元素代表一种商品的价格,你可以购买2种商品,但是总价格不能超过"target"值。
代码解析如下:
-
首先,对输入的"nums"列表进行排序,这样可以从最小的价格开始考虑,减少后续搜索的范围。
-
初始化计数器"count"为0,用于记录满足条件的购买方案的数量。
-
使用一个外循环遍历排序后的"nums"列表,对于每一个价格,尝试通过添加后续的商品价格来接近或达到"target"值。
- 计算"target"与当前商品价格的差值"n"。
- 如果这个差值"n"在列表中存在(使用一个内循环来查找),那么存在后续满足条件的可能性。在内循环中,如果找到一个商品价格小于等于差值"n",就将一个标志位"flag"设为True。当找到一个大于差值"n"的商品价格时,就跳出内循环。
- 如果内循环结束后,"flag"仍为True,说明找到了一个满足条件的商品组合。如果最后一个商品价格也小于等于差值"n",那么购买方案的数量为外循环的索引"i"到内循环结束时的索引"j",所以将"(j-i)"加到计数器"count"上;否则,购买方案的数量为外循环的索引"i"到内循环结束时的索引"j-1",所以将"(j-i-1)"加到计数器"count"上。
-
最后,返回计数器"count"的值,即满足条件的购买方案的数量。
3.python代码
class Solution:
def purchasePlans(self, nums: list[int], target: int) -> int:
nums.sort()
count=0
for i in range(len(nums)-1):
n=target-nums[i]
flag=False
for j in range(i+1,len(nums)):
if nums[j]<=n:
flag=True
else:
break
if flag:
if j==len(nums)-1 and nums[j]<=n:
count+=j-i
else:
count+=j-i-1
return count