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 = 0for i in range(n - 3):# 避免重复计数if i > 0 and nums[i] == nums[i - 1]:continuefor j in range(i + 1, n - 2):# 避免重复计数if j > i + 1 and nums[j] == nums[j - 1]:continuefor p in range(j + 1, n - 1):# 避免重复计数if p > j + 1 and nums[p] == nums[p - 1]:continuefor q in range(p + 1, n):# 计算乘积并判断是否满足条件product = nums[i] * nums[j] * nums[p] * nums[q]if product <= k:count += 1return count
print(count_combinations([1, 1, 1, 2, 2], 3))