计算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