首页 文章资讯内容详情

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

2026-06-03 1 花语

假设我们有一个名为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