Python程序：查找优质三元组的数量

假设我们有一个数组nums，以及三个不同的整数a、b和c。我们需要找到优质三元组的数量。如果满足以下条件，则三元组(nums[i]，nums[j]，nums[k])被称为优质三元组：

• 0 <= i < j < k < nums数组元素个数

• |nums[i] - nums[j]| <= a

• |nums[j] - nums[k]| <= b

• |nums[i] - nums[k]| <= c

我们需要计算优质三元组的数量。

因此，如果输入类似于nums= [5,2,3,3,12,9]，a = 7，b = 2，c = 3，则输出将为4，因为优质三元组为[(5,2,3), (5,2,3), (5,3,3), (2,3,3)]

为了解决这个问题，我们将遵循以下步骤：

• res := 0

• 对于 i 从 0 到 nums 大小 - 1，执行：

  • 对于 j 从 i+1 到 nums 大小 - 1，执行：

    • 对于 k 从 j+1 到 nums 大小 - 1，执行：

      • 如果 |nums[i] - nums[j]| <= a 且 |nums[j] - nums[k]| <= b 且 |nums[i] - nums[k]| <= c，则

        • res := res + 1

• 返回 res

示例 (Python)

让我们来看下面的实现，以便更好地理解：

 在线演示

def solve(nums, a, b, c):
   res = 0
   for i in range(len(nums)):
      for j in range(i+1, len(nums)):
         for k in range(j+1, len(nums)):
            if abs(nums[i] - nums[j]) <= a and abs(nums[j] - nums[k]) <= b and abs(nums[i] - nums[k]) <= c:
               res += 1
   return res

nums= [5,2,3,3,12,9]
a = 7
b = 2
c = 3
print(solve(nums, a, b, c))

输入

[5,2,3,3,12,9], 7, 2 3

输出

4

Arnab Chakraborty

更新于：2021年5月17日

2K+ 次浏览

启动你的职业生涯

通过完成课程获得认证

开始学习