计算Python中数组元素相同的索引对的程序
假设我们有一个名为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