计算Python中数组元素相同的索引对的程序

• Z时代
• 2024-01-10
• 分类：综合

假设我们有一个名为 nums 的数字列表。我们必须找到使 nums[i] 和 nums[j] 相同的对 i < j 的数量。

因此，如果输入类似于 nums = [5, 4, 5, 4, 4]，那么输出将是 4，因为我们有像 (0, 2), (1, 3), (1, 4) 这样的索引对) 和 (3, 4)。

示例

让我们看下面的实现来更好地理解

from collections import Counter
def solve(nums):
   c = Counter(nums)
   count = 0
   for n in c.values():
      count += n * (n - 1) // 2
   return count
nums = [5, 4, 5, 4, 4]
print(solve(nums))

输入

[5, 4, 5, 4, 4]

4

以上是 计算Python中数组元素相同的索引对的程序 的全部内容， 来源链接： utcz.com/z/343698.html