Python算法例7 四数乘积

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))