1. 问题描述
给定一个长度为n的数组a和一个正整数k,从数组中选择四个数,要求四个数的乘积小于等于k,求方案总数。
2. 问题示例
给定n=5,a=[1,1,1,2,2],k=3,返回2。
3.代码实现
# 使用嵌套循环的方式来求解。首先,我们可以将数组a排序,
# 然后使用四重循环遍历所有可能的四个数的组合。
# 在每次循环中,我们计算四个数的乘积,并将符合条件(小于等于k)的组合计数。
def count_combinations(nums, k):
nums.sort() # 将数组排序
n = len(nums)
count = 0
for i in range(n - 3):
# 避免重复计数
if i > 0 and nums[i] == nums[i - 1]:
continue
for j in range(i + 1, n - 2):
# 避免重复计数
if j > i + 1 and nums[j] == nums[j - 1]:
continue
for p in range(j + 1, n - 1):
# 避免重复计数
if p > j + 1 and nums[p] == nums[p - 1]:
continue
for q in range(p + 1, n):
# 计算乘积并判断是否满足条件
product = nums[i] * nums[j] * nums[p] * nums[q]
if product <= k:
count += 1
return count
print(count_combinations([1, 1, 1, 2, 2], 3))